Skip to content

Commit 45bfe0c

Browse files
Update ext/OptimizationEnzymeExt.jl
Co-authored-by: Vaibhav Kumar Dixit <[email protected]>
1 parent 99184f7 commit 45bfe0c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ext/OptimizationEnzymeExt.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -170,7 +170,7 @@ function Optimization.instantiate_function(f::OptimizationFunction{true},
170170

171171
if f.hess === nothing
172172
function g(θ, bθ, f, p, args...)
173-
Enzyme.autodiff_deferred(Enzyme.Reverse, Const(firstapply), Const(f),
173+
Enzyme.autodiff_deferred(Enzyme.Reverse, Const(firstapply), Active, Const(f),
174174
Enzyme.Duplicated(θ, bθ),
175175
Const(p),
176176
args...)

0 commit comments

Comments
 (0)