fix Enzyme call

TuringLang · Aug 9, 2024 · 4207a87 · 4207a87
1 parent c2a014b
commit 4207a87
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ext/AdvancedVIEnzymeExt.jl b/ext/AdvancedVIEnzymeExt.jl
@@ -17,7 +17,7 @@ function AdvancedVI.value_and_gradient!(
     ∇θ = DiffResults.gradient(out)
     fill!(∇θ, zero(T))
     _, y = Enzyme.autodiff(
-        Enzyme.ReverseWithPrimal, f, Enzyme.Active, Enzyme.Duplicated(θ, ∇θ)
+        Enzyme.ReverseWithPrimal, Enzyme.Const(f), Enzyme.Active, Enzyme.Duplicated(θ, ∇θ)
     )
     DiffResults.value!(out, y)
     return out