Fix enzyme_nocache (#2146)

* Fix enzyme_nocache * fix * fix
EnzymeAD · Nov 3, 2024 · b6ab325 · b6ab325
1 parent 5ba7dcb
commit b6ab325
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/enzyme/Enzyme/EnzymeLogic.cpp b/enzyme/Enzyme/EnzymeLogic.cpp
@@ -270,7 +270,7 @@ struct CacheAnalysis {
       return false;
     }
 
-    if (hasMetadata(&li, "enzyme_force_recompute"))
+    if (hasNoCache(&li))
       return false;
 
     if (EnzymeJuliaAddrLoad)

diff --git a/enzyme/Enzyme/Utils.h b/enzyme/Enzyme/Utils.h
@@ -1181,6 +1181,9 @@ static inline bool hasNoCache(llvm::Value *op) {
         return true;
     }
   }
+  if (auto I = dyn_cast<Instruction>(op))
+    if (hasMetadata(I, "enzyme_nocache"))
+      return true;
   return false;
 }