Skip to content

Commit 60aada0

Browse files
revert enzyme choice on callbacks
1 parent ab5469d commit 60aada0

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

src/sensitivity_interface.jl

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -365,12 +365,14 @@ function adjoint_sensitivities(sol, args...;
365365
has_cb = false
366366
end
367367

368-
_sensealg = if isinplace(sol.prob)
369-
setvjp(sensealg, inplace_vjp(sol.prob, sol.prob.u0, sol.prob.p, verbose))
370-
elseif has_cb
371-
setvjp(sensealg, ReverseDiffVJP())
368+
if !has_cb
369+
_sensealg = if isinplace(sol.prob)
370+
setvjp(sensealg, inplace_vjp(sol.prob, sol.prob.u0, sol.prob.p, verbose))
371+
else
372+
setvjp(sensealg, ZygoteVJP())
373+
end
372374
else
373-
setvjp(sensealg, ZygoteVJP())
375+
_sensealg = setvjp(sensealg, ReverseDiffVJP())
374376
end
375377

376378
return try

0 commit comments

Comments
 (0)