fix

EnzymeAD · Sep 28, 2024 · 3ee870b · 3ee870b
1 parent a4bbf20
commit 3ee870b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/internal_rules.jl b/src/internal_rules.jl
@@ -351,6 +351,8 @@ function EnzymeRules.augmented_primal(
         EnzymeRules.overwritten(config)[2:end],
         InlineABI,
         false,
+        false,
+        false
     }()
     fwd_thunk, rev_thunk =
         autodiff_thunk(config2, BodyTy, Const, typeof(count), map(typeof, args)...)
@@ -405,6 +407,8 @@ function EnzymeRules.reverse(
         EnzymeRules.overwritten(config)[2:end],
         InlineABI,
         false,
+        false,
+        false
     }()
     fwd_thunk, rev_thunk =
         autodiff_thunk(config2, BodyTy, Const, typeof(count), map(typeof, args)...)