[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:04 PST 2024


================
@@ -6100,6 +6100,8 @@ NVPTXTargetLowering::shouldExpandAtomicRMWInIR(AtomicRMWInst *AI) const {
 
   if (AI->isFloatingPointOperation()) {
     if (AI->getOperation() == AtomicRMWInst::BinOp::FAdd) {
+      if (Ty->isHalfTy() && STI.getSmVersion() >= 70)
+        return AtomicExpansionKind::None;
----------------
Artem-B wrote:

It also requires PTX ISA 6.3.

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


More information about the llvm-commits mailing list