fix

EnzymeAD · Nov 4, 2024 · 94dc476 · 94dc476
1 parent 2b0dd8e
commit 94dc476
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/enzyme/Enzyme/JLInstSimplify.cpp b/enzyme/Enzyme/JLInstSimplify.cpp
@@ -224,8 +224,9 @@ bool jlInstSimplify(llvm::Function &F, TargetLibraryInfo &TLI,
         }
         auto llhs = dyn_cast<LoadInst>(lhs);
         auto lrhs = dyn_cast<LoadInst>(rhs);
-        if (llhs && lrhs && isa<PointerType>(llhs->getType()) &&
-            isa<PointerType>(lrhs->getType())) {
+        if (llhs && lrhs && 
+            (isa<PointerType>(llhs->getType()) || isa<IntegerType>(llhs->getType())) &&
+            (isa<PointerType>(lrhs->getType()) || isa<IntegerType>(lrhs->getType()))) {
           auto lhsv =
               getBaseObject(llhs->getOperand(0), /*offsetAllowed*/ false);
           auto rhsv =