Skip to content

Commit 72d8027

Browse files
jClugstorChrisRackauckas
authored andcommitted
fix jac and grad resizing
1 parent b74a9bf commit 72d8027

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/integrators/interface.jl

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -203,8 +203,10 @@ function DiffEqBase.resize_non_user_cache!(integrator::DDEIntegrator,
203203
cache::RosenbrockMutableCache, i)
204204
cache.J = similar(cache.J, i, i)
205205
cache.W = similar(cache.W, i, i)
206-
OrdinaryDiffEqDifferentiation.resize_jac_config!(cache.jac_config, i)
207-
OrdinaryDiffEqDifferentiation.resize_grad_config!(cache.grad_config, i)
206+
207+
208+
OrdinaryDiffEqDifferentiation.resize_jac_config!(cache.uf, cache.du1, cache.jac_config, OrdinaryDiffEqCore.alg_autodiff(integrator.alg), integrator.u)
209+
OrdinaryDiffEqDifferentiation.resize_grad_config!(cache.uf, cache.du1, cache.grad_config, OrdinaryDiffEqCore.alg_autodiff(integrator.alg), integrator.u)
208210
nothing
209211
end
210212

0 commit comments

Comments
 (0)