Fix combined temp cache for reverse (#2131)

EnzymeAD · Oct 22, 2024 · c994307 · c994307
1 parent 6185e17
commit c994307
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/enzyme/Enzyme/AdjointGenerator.h b/enzyme/Enzyme/AdjointGenerator.h
@@ -5489,8 +5489,11 @@ class AdjointGenerator : public llvm::InstVisitor<AdjointGenerator> {
             It.setHeadBit(true);
             BuilderZ.SetInsertPoint(It);
 #endif
-            cachereplace = BuilderZ.CreatePHI(call.getType(), 1,
-                                              call.getName() + "_tmpcacheB");
+            if (Mode == DerivativeMode::ReverseModeCombined)
+              cachereplace = newCall;
+            else
+              cachereplace = BuilderZ.CreatePHI(call.getType(), 1,
+                                                call.getName() + "_tmpcacheB");
             cachereplace = gutils->cacheForReverse(
                 BuilderZ, cachereplace,
                 getIndex(&call, CacheType::Self, BuilderZ));