[llvm] [NVPTX] Add support for atomic add for f16 type (PR #84295)

Artem Belevich via llvm-commits llvm-commits at lists.llvm.org
Thu Mar 7 10:22:06 PST 2024


================
@@ -1630,6 +1630,13 @@ defm INT_PTX_ATOM_ADD_GEN_64 : F_ATOMIC_2<i64, Int64Regs, "", ".u64", ".add",
 defm INT_PTX_ATOM_ADD_GEN_64_USE_G : F_ATOMIC_2<i64, Int64Regs, ".global", ".u64",
   ".add", atomic_load_add_64_gen, i64imm, imm>;
 
+defm INT_PTX_ATOM_ADD_G_F16 : F_ATOMIC_2<f16, Int16Regs, ".global", ".f16", ".add.noftz",
+  atomic_load_add_g, f16imm, fpimm, [hasSM<70>]>;
----------------
Artem-B wrote:

Needs `hasPTX<63>` here, too.

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


More information about the llvm-commits mailing list