fix rosenbrock addstep

Author: oscardssmith

lib/OrdinaryDiffEqRosenbrock/src/stiff_addsteps.jl

@@ -63,7 +63,7 @@ function _ode_addsteps!(k, t, uprev, u, dt, f, p,
 
         ### Jacobian does not need to be re-evaluated after an event
         ### Since it's unchanged
-        jacobian2W!(W, mass_matrix, dtγ, J, true)
+        jacobian2W!(W, mass_matrix, dtγ, J)
 
         linsolve = cache.linsolve
 
@@ -215,7 +215,7 @@ function _ode_addsteps!(
 
         ### Jacobian does not need to be re-evaluated after an event
         ### Since it's unchanged
-        jacobian2W!(W, mass_matrix, dtgamma, J, true)
+        jacobian2W!(W, mass_matrix, dtgamma, J)
 
         linsolve = cache.linsolve
 
@@ -394,7 +394,7 @@ function _ode_addsteps!(k, t, uprev, u, dt, f, p, cache::RosenbrockCache,
         @.. linsolve_tmp = @muladd fsalfirst + dtgamma * dT
 
         # Jacobian does not need to be re-evaluated after an event since it's unchanged
-        jacobian2W!(W, mass_matrix, dtgamma, J, true)
+        jacobian2W!(W, mass_matrix, dtgamma, J)
 
         linsolve = cache.linsolve
 
@@ -623,7 +623,7 @@ function _ode_addsteps!(k, t, uprev, u, dt, f, p, cache::Rosenbrock5Cache,
 
         ### Jacobian does not need to be re-evaluated after an event
         ### Since it's unchanged
-        jacobian2W!(W, mass_matrix, dtgamma, J, true)
+        jacobian2W!(W, mass_matrix, dtgamma, J)
 
         linsolve = cache.linsolve
 

test/interface/utility_tests.jl

@@ -39,7 +39,7 @@ using OrdinaryDiffEq.OrdinaryDiffEqDifferentiation: WOperator, calc_W, calc_W!,
 
     # But jacobian2W! will update the cache
     jacobian2W!(integrator.cache.nlsolver.cache.W._concrete_form, mm,
-        dtgamma, integrator.cache.nlsolver.cache.W.J.A, false)
+        dtgamma, integrator.cache.nlsolver.cache.W.J.A)
     @test convert(AbstractMatrix, integrator.cache.nlsolver.cache.W) == concrete_W
     ldiv!(tmp, lu!(integrator.cache.nlsolver.cache.W), u0)
     @test tmp == concrete_W \ u0