diff --git a/mlir/lib/Conversion/GPUToROCDL/LowerGpuOpsToROCDLOps.cpp b/mlir/lib/Conversion/GPUToROCDL/LowerGpuOpsToROCDLOps.cpp index 489891e06fcb..c321d8e3badb 100644 --- a/mlir/lib/Conversion/GPUToROCDL/LowerGpuOpsToROCDLOps.cpp +++ b/mlir/lib/Conversion/GPUToROCDL/LowerGpuOpsToROCDLOps.cpp @@ -72,7 +72,7 @@ struct LowerGpuOpsToROCDLOpsPass target.addIllegalDialect(); target.addIllegalOp(); + LLVM::Log2Op, LLVM::SinOp, LLVM::TanhOp>(); target.addIllegalOp(); target.addLegalDialect(); // TODO: Remove once we support replacing non-root ops. @@ -113,6 +113,8 @@ void mlir::populateGpuToROCDLConversionPatterns( "__ocml_log10_f64"); patterns.insert>(converter, "__ocml_log2_f32", "__ocml_log2_f64"); + patterns.insert>(converter, "__ocml_sin_f32", + "__ocml_sin_f64"); patterns.insert>(converter, "__ocml_tanh_f32", "__ocml_tanh_f64"); }