fix

EnzymeAD · Dec 23, 2024 · 50ee0f4 · 50ee0f4
1 parent 3edebbe
commit 50ee0f4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/enzyme/Enzyme/GradientUtils.cpp b/enzyme/Enzyme/GradientUtils.cpp
@@ -3822,6 +3822,9 @@ bool GradientUtils::legalRecompute(const Value *val,
     }
   }
 
+  if (isa<AtomicRMWInst>(val))
+    return false;
+
   if (auto phi = dyn_cast<PHINode>(val)) {
     if (auto uiv = hasUninverted(val)) {
       if (auto dli = dyn_cast_or_null<LoadInst>(uiv)) {