Merge pull request #2275 from oscardssmith/os/fix-default-interp

fix default algorithm interpolation
SciML · Jul 5, 2024 · b03a2a3 · b03a2a3
2 parents 6c0bce5 + 84b1b71
commit b03a2a3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/dense/generic_dense.jl b/src/dense/generic_dense.jl
@@ -595,6 +595,8 @@ function ode_interpolation!(vals, tvals, id::I, idxs, deriv::D, p,
     if cache isa CompositeCache
         current_alg = id.alg_choice[i₊]
         cache_i₊ = cache.caches[current_alg]
+    elseif cache isa DefaultCache
+        current_alg = id.alg_choice[i₊]
     else
         cache_i₊ = cache
     end

diff --git a/test/interface/default_solver_tests.jl b/test/interface/default_solver_tests.jl
@@ -11,6 +11,8 @@ tsitsol = solve(prob_ode_2Dlinear, Tsit5())
 @test sol.stats.nf == tsitsol.stats.nf
 @test all(isequal(1), sol.alg_choice)
 @test sol(0.5) == only(sol([0.5]).u) == tsitsol(0.5)
+x = [zeros(4,2) for _ in 1:5]
+@test sol(x, 0:.1:.4) == tsitsol(x, 0:.1:.4)
 
 sol = solve(prob_ode_2Dlinear, reltol = 1e-10)
 vernsol = solve(prob_ode_2Dlinear, Vern7(), reltol = 1e-10)