Skip to content

Commit 0daefce

Browse files
Merge pull request SciML#610 from ArnoStrouwen/enzyme
make sensitivity_solution work with Enzyme
2 parents 0c62711 + a165c34 commit 0daefce

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/solutions/ode_solutions.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -399,9 +399,9 @@ function sensitivity_solution(sol::ODESolution, u, t)
399399
end
400400

401401
interp = enable_interpolation_sensitivitymode(sol.interp)
402-
403-
ODESolution{T, N}(u, sol.u_analytic, sol.errors, t,
404-
nothing, sol.prob,
402+
ODESolution{T, N}(u, sol.u_analytic, sol.errors,
403+
t isa Vector ? t : collect(t),
404+
sol.k, sol.prob,
405405
sol.alg, interp,
406406
sol.dense, sol.tslocation,
407407
sol.stats, sol.alg_choice, sol.retcode)

0 commit comments

Comments
 (0)