diff --git a/mlir/lib/Conversion/GPUToNVVM/LowerGpuOpsToNVVMOps.cpp b/mlir/lib/Conversion/GPUToNVVM/LowerGpuOpsToNVVMOps.cpp index d64c4d64cad84..70c97b3566662 100644 --- a/mlir/lib/Conversion/GPUToNVVM/LowerGpuOpsToNVVMOps.cpp +++ b/mlir/lib/Conversion/GPUToNVVM/LowerGpuOpsToNVVMOps.cpp @@ -383,16 +383,14 @@ struct LowerGpuOpsToNVVMOpsPass final LLVMConversionTarget target(getContext()); // Set higher benefit, so patterns will run before generic LLVM lowering. + // Make sure the benefit here is higher than ArithToLLVMDialectInterface and + // MathToLLVMDialectInterface. populateGpuToNVVMConversionPatterns(converter, llvmPatterns, /*benefit=*/10); llvm::SmallDenseSet allowedDialectsSet(allowedDialects.begin(), allowedDialects.end()); for (Dialect *dialect : getContext().getLoadedDialects()) { - // Skip math patterns as nvvm needs custom math lowering. - if (isa(dialect)) - continue; - bool allowed = allowedDialectsSet.contains(dialect->getNamespace()); // Empty `allowedDialectsSet` means all dialects are allowed. if (!allowedDialectsSet.empty() && !allowed) diff --git a/mlir/test/Conversion/GPUToNVVM/gpu-to-generic-llvm.mlir b/mlir/test/Conversion/GPUToNVVM/gpu-to-generic-llvm.mlir new file mode 100644 index 0000000000000..4594025bdfc90 --- /dev/null +++ b/mlir/test/Conversion/GPUToNVVM/gpu-to-generic-llvm.mlir @@ -0,0 +1,20 @@ +// RUN: mlir-opt %s -convert-gpu-to-nvvm -split-input-file | FileCheck %s + +/// Math/arith ops that are not supported by libdevice +/// should be converted by generic LLVM lowering patterns. + +gpu.module @generic_llvm_test_module_0 { + // CHECK-LABEL: @math_abs_non_i32 + func.func @math_abs_non_i32(%arg_i64: i64, %arg_i16: i16, %arg_i8: i8, %arg_i1: i1) + -> (i64, i16, i8, i1) { + // CHECK: "llvm.intr.abs"{{.*}} : (i64) -> i64 + %abs_i64 = math.absi %arg_i64 : i64 + // CHECK: "llvm.intr.abs"{{.*}} : (i16) -> i16 + %abs_i16 = math.absi %arg_i16 : i16 + // CHECK: "llvm.intr.abs"{{.*}} : (i8) -> i8 + %abs_i8 = math.absi %arg_i8 : i8 + // CHECK: "llvm.intr.abs"{{.*}} : (i1) -> i1 + %abs_i1 = math.absi %arg_i1 : i1 + return %abs_i64, %abs_i16, %abs_i8, %abs_i1 : i64, i16, i8, i1 + } +}