Fix unwrap of invert (rust-lang#541)

Co-authored-by: Valentin Churavy <v.churavy@gmail.com>
EnzymeAD · Mar 3, 2022 · 1c331bc · 1c331bc
1 parent 5093a8c
commit 1c331bc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/enzyme/Enzyme/GradientUtils.cpp b/enzyme/Enzyme/GradientUtils.cpp
@@ -798,8 +798,8 @@ Value *GradientUtils::unwrapM(Value *const val, IRBuilder<> &BuilderM,
         if (isOriginalBlock(*BuilderM.GetInsertBlock())) {
           pidx = invertPointerM(dli->getOperand(0), BuilderM);
         } else {
-          pidx =
-              lookupM(invertPointerM(dli->getOperand(0), BuilderM), BuilderM);
+          pidx = lookupM(invertPointerM(dli->getOperand(0), BuilderM), BuilderM,
+                         available);
         }
 
         if (pidx == nullptr)