fix resize

Author: oscardssmith

lib/OrdinaryDiffEqRosenbrock/src/rosenbrock_caches.jl

@@ -8,7 +8,7 @@ get_fsalfirstlast(cache::RosenbrockMutableCache, u) = (zero(u), zero(u))
 
 # Shampine's Low-order Rosenbrocks
 
-@cache mutable struct RosenbrockCache{uType, rateType, uNoUnitsType, JType, WType, TabType,
+mutable struct RosenbrockCache{uType, rateType, uNoUnitsType, JType, WType, TabType,
     TFType, UFType, F, JCType, GCType, RTolType, A, StepLimiter, StageLimiter} <: RosenbrockMutableCache
     u::uType
     uprev::uType
@@ -37,6 +37,10 @@ get_fsalfirstlast(cache::RosenbrockMutableCache, u) = (zero(u), zero(u))
     step_limiter!::StepLimiter
     stage_limiter!::StageLimiter
 end
+function full_cache(c::RosenbrockCache)
+    return [c.u, c.uprev, c.dense..., c.du, c.du1, c.du2,
+            c.ks..., c.fsalfirst, c.fsallast, c.dT, c.tmp, c.atmp, c.weight, c.linsolve_tmp]
+end
 
 @cache mutable struct Rosenbrock23Cache{uType, rateType, uNoUnitsType, JType, WType,
     TabType, TFType, UFType, F, JCType, GCType,

lib/OrdinaryDiffEqRosenbrock/src/rosenbrock_perform_step.jl

@@ -1328,7 +1328,7 @@ end
             solverdata = (; gamma = dtgamma))
     end
 
-    @.. broadcast=false $(_vec(ks[1]))=-linres.u
+    @.. $(_vec(ks[1]))=-linres.u
     integrator.stats.nsolve += 1
 
     for stage in 2:length(ks)

lib/OrdinaryDiffEqRosenbrock/src/stiff_addsteps.jl

@@ -383,8 +383,7 @@ function _ode_addsteps!(k, t, uprev, u, dt, f, p, cache::RosenbrockCache,
 
         linsolve = cache.linsolve
 
-        linres = dolinsolve(cache, linsolve; A = W, b = _vec(linsolve_tmp),
-            reltol = cache.reltol)
+        linres = dolinsolve(cache, linsolve; A = W, b = _vec(linsolve_tmp), reltol = cache.reltol)
         @.. $(_vec(ks[1]))=-linres.u
         for stage in 2:length(ks)
             tmp .= uprev