Non-power of two cache switch (#2204)

EnzymeAD · Dec 23, 2024 · 79ac406 · 79ac406
1 parent f5767ef
commit 79ac406
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/enzyme/Enzyme/Utils.cpp b/enzyme/Enzyme/Utils.cpp
@@ -102,6 +102,10 @@ llvm::cl::opt<bool> EnzymeMemmoveWarning(
 llvm::cl::opt<bool> EnzymeRuntimeError(
     "enzyme-runtime-error", cl::init(false), cl::Hidden,
     cl::desc("Emit Runtime errors instead of compile time ones"));
+
+llvm::cl::opt<bool> EnzymeNonPower2Cache(
+    "enzyme-non-power2-cache", cl::init(false), cl::Hidden,
+    cl::desc("Disable caching of integers which are not a power of 2"));
 }
 
 void ZeroMemory(llvm::IRBuilder<> &Builder, llvm::Type *T, llvm::Value *obj,

diff --git a/enzyme/Enzyme/Utils.h b/enzyme/Enzyme/Utils.h
@@ -91,6 +91,7 @@ enum class ErrorType {
 extern "C" {
 /// Print additional debug info relevant to performance
 extern llvm::cl::opt<bool> EnzymePrintPerf;
+extern llvm::cl::opt<bool> EnzymeNonPower2Cache;
 extern llvm::cl::opt<bool> EnzymeStrongZero;
 extern llvm::cl::opt<bool> EnzymeBlasCopy;
 extern llvm::cl::opt<bool> EnzymeLapackCopy;
@@ -1194,6 +1195,10 @@ static inline bool hasNoCache(llvm::Value *op) {
       }
     }
   }
+  if (auto IT = dyn_cast<IntegerType>(op->getType()))
+    if (!isPowerOf2_64(IT->getBitWidth()) && !EnzymeNonPower2Cache)
+      return true;
+
   return false;
 }