[Mlir-commits] [mlir] [mlir][gpu] Support `minimumnumf` and `maximumnumf` in GPU dialect (PR #226903)

lorenzo chelini llvmlistbot at llvm.org
Mon Sep 28 01:57:33 PDT 2026


================
@@ -0,0 +1,21 @@
+// RUN: mlir-opt --convert-gpu-to-nvvm %s | FileCheck %s
+
+gpu.module @kernels {
+  // CHECK-LABEL: llvm.func @minimum_number
+  // CHECK: llvm.intr.minimumnum
+  // CHECK: llvm.return
+  gpu.func @minimum_number(%arg0 : f32, %out : memref<f32>) kernel {
+    %result = gpu.all_reduce minimumnumf %arg0 uniform {} : (f32) -> f32
+    memref.store %result, %out[] : memref<f32>
+    gpu.return
+  }
+
+  // CHECK-LABEL: llvm.func @maximum_number
+  // CHECK: llvm.intr.maximumnum
+  // CHECK: llvm.return
+  gpu.func @maximum_number(%arg0 : f32, %out : memref<f32>) kernel {
----------------
chelini wrote:

Could you please also add positive tests for `subgroup_reduce`?

https://github.com/llvm/llvm-project/pull/226903


More information about the Mlir-commits mailing list