Fix caching for forwarderr

EnzymeAD · Apr 15, 2024 · a5a40bc · a5a40bc
1 parent 1f62a0d
commit a5a40bc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/enzyme/Enzyme/CallDerivatives.cpp b/enzyme/Enzyme/CallDerivatives.cpp
@@ -3195,7 +3195,8 @@ bool AdjointGenerator::handleKnownCallDerivatives(
     }
 
     // Cache and rematerialization irrelevant for forward mode.
-    if (Mode == DerivativeMode::ForwardMode) {
+    if (Mode == DerivativeMode::ForwardMode ||
+        Mode == DerivativeMode::ForwardModeError) {
       eraseIfUnused(call);
       return true;
     }