[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