diff --git a/mlir/lib/Conversion/GPUToROCDL/LowerGpuOpsToROCDLOps.cpp b/mlir/lib/Conversion/GPUToROCDL/LowerGpuOpsToROCDLOps.cpp --- a/mlir/lib/Conversion/GPUToROCDL/LowerGpuOpsToROCDLOps.cpp +++ b/mlir/lib/Conversion/GPUToROCDL/LowerGpuOpsToROCDLOps.cpp @@ -99,6 +99,10 @@ GPUFuncOpLowering<5>, GPUReturnOpLowering>(converter); patterns.insert>(converter, "__ocml_fabs_f32", "__ocml_fabs_f64"); + patterns.insert>(converter, "__ocml_atan_f32", + "__ocml_atan_f64"); + patterns.insert>(converter, "__ocml_atan2_f32", + "__ocml_atan2_f64"); patterns.insert>(converter, "__ocml_ceil_f32", "__ocml_ceil_f64"); patterns.insert>(converter, "__ocml_cos_f32",