[llvm] [IR][AtomicExpand] Add elementwise modifier to atomicrmw; automatically expand for all targets (PR #189517)
Yonah Goldberg via llvm-commits
llvm-commits at lists.llvm.org
Thu Apr 9 09:49:37 PDT 2026
================
@@ -265,6 +268,23 @@ static bool atomicSizeSupported(const TargetLowering *TLI, Inst *I) {
Size <= TLI->getMaxAtomicSizeInBitsSupported() / 8;
}
+/// Returns true if we can lower atomicrmw elementwise using normal atomicrmw.
+bool AtomicExpandImpl::canReuseWholeValueAtomicRMW(AtomicRMWInst *AI) {
+ assert(AI->isElementwise() && "expected elementwise atomicrmw");
+
+ // Integer non-elementwise vector atomicrmw is illegal IR, so we need to be
+ // careful to reject these before removing the elementwise modifier.
+ if (!AI->isFloatingPointOperation())
+ return false;
+
+ AI->setElementwise(false);
----------------
YonahGoldberg wrote:
Oh I see, yeah that actually makes more sense I agree.
https://github.com/llvm/llvm-project/pull/189517
More information about the llvm-commits
mailing list