[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