Merge pull request #342 from CliMA/tr/fix-prob-remake

Remake problem manually during multirate init

Author: charleskawczynski

src/solvers/multirate.jl

@@ -24,16 +24,32 @@ struct MultirateCache{OC, II}
     innerinteg::II
 end
 
+"""
+    cts_remake(prob::DiffEqBase.AbstractODEProblem; f::DiffEqBase.AbstractODEFunction)
+
+Remake an ODE problem with a new function `f`.
+"""
+function cts_remake(prob::DiffEqBase.AbstractODEProblem; f::DiffEqBase.AbstractODEFunction)
+    return DiffEqBase.ODEProblem{DiffEqBase.isinplace(prob)}(
+        f,
+        prob.u0,
+        prob.tspan,
+        prob.p,
+        prob.problem_type;
+        prob.kwargs...,
+    )
+end
+
 function init_cache(prob::DiffEqBase.AbstractODEProblem, alg::Multirate; dt, fast_dt, kwargs...)
 
     @assert prob.f isa DiffEqBase.SplitFunction
 
     # subproblems
-    outerprob = DiffEqBase.remake(prob; f = prob.f.f2)
+    outerprob = cts_remake(prob; f = prob.f.f1)
     outercache = init_cache(outerprob, alg.slow)
 
     innerfun = init_inner(prob, outercache, dt)
-    innerprob = DiffEqBase.remake(prob; f = innerfun)
+    innerprob = cts_remake(prob; f = innerfun)
     innerinteg = DiffEqBase.init(innerprob, alg.fast; dt = fast_dt, kwargs...)
     return MultirateCache(outercache, innerinteg)
 end