[Mlir-commits] [mlir] Allow 16 bit floating point operand for LLVM_AtomicRMWOp (PR #110553)

Giuseppe Rossini llvmlistbot at llvm.org
Wed Oct 2 01:21:48 PDT 2024


================
@@ -1496,50 +1496,53 @@ llvm.func @elements_constant_3d_array() -> !llvm.array<2 x array<2 x array<2 x i
 // CHECK-LABEL: @atomicrmw
 llvm.func @atomicrmw(
     %f32_ptr : !llvm.ptr, %f32 : f32,
+    %f16_vec_ptr : !llvm.ptr, %f16_vec : vector<2xf16>,
     %i32_ptr : !llvm.ptr, %i32 : i32) {
   // CHECK: atomicrmw fadd ptr %{{.*}}, float %{{.*}} monotonic
   %0 = llvm.atomicrmw fadd %f32_ptr, %f32 monotonic : !llvm.ptr, f32
----------------
giuseros wrote:

Also, minor nitpick, could you move this to be the last test? In this way you don't have to renumber all the SSAs

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


More information about the Mlir-commits mailing list