fix

EnzymeAD · Nov 4, 2024 · 2b0dd8e · 2b0dd8e
1 parent 2336eec
commit 2b0dd8e
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/enzyme/Enzyme/JLInstSimplify.cpp b/enzyme/Enzyme/JLInstSimplify.cpp
@@ -122,6 +122,10 @@ bool notCapturedBefore(llvm::Value *V, Instruction *inst) {
       if (isa<CmpInst>(UI)) {
         continue;
       }
+      if (isa<SelectInst>(UI) || isa<PHINode>(UI)) {
+        todo.push_back(UI);
+        continue;
+      }
       if (isa<LoadInst>(UI)) {
         todo.push_back(UI);
         continue;
@@ -162,7 +166,9 @@ bool jlInstSimplify(llvm::Function &F, TargetLibraryInfo &TLI,
       if (auto LI = dyn_cast<LoadInst>(&I)) {
         size_t offset = 0;
         auto obj = getBaseObject(LI->getPointerOperand(),
-                                 /*offsetAllowed=*/false, &offset);
+                                 /*offsetAllowed=*/true, &offset);
+        // llvm::errs() << " LI: " << *LI << " obj: " << *obj << " off " <<
+        // offset << " ptr: " << *LI->getPointerOperand() << "\n";
         if (auto CI = dyn_cast<CallBase>(obj)) {
           if (getFuncNameFromCall(CI) == "jl_alloc_genericmemory") {
             if (offset == 0) {