Skip to content

Commit 33a57d2

Browse files
Allow for Enzyme choice with callbacks automatically
1 parent 2a110dc commit 33a57d2

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

Diff for: src/sensitivity_interface.jl

+4-7
Original file line numberDiff line numberDiff line change
@@ -364,14 +364,11 @@ function adjoint_sensitivities(sol, args...;
364364
else
365365
has_cb = false
366366
end
367-
if !has_cb
368-
_sensealg = if isinplace(sol.prob)
369-
setvjp(sensealg, inplace_vjp(sol.prob, sol.prob.u0, sol.prob.p, verbose))
370-
else
371-
setvjp(sensealg, ZygoteVJP())
372-
end
367+
368+
_sensealg = if isinplace(sol.prob)
369+
setvjp(sensealg, inplace_vjp(sol.prob, sol.prob.u0, sol.prob.p, verbose))
373370
else
374-
_sensealg = setvjp(sensealg, ReverseDiffVJP())
371+
setvjp(sensealg, ZygoteVJP())
375372
end
376373

377374
return try

0 commit comments

Comments
 (0)