simplify

Author: oscarddssmith

src/problems/ode_problems.jl

@@ -200,7 +200,7 @@ function ODEProblem(f, u0, tspan, p = NullParameters(); kwargs...)
     _tspan = promote_tspan(tspan)
     _f = if iip
         out = copy(u0) # TODO: do this properly
-        ODEFunction{iip, FullSpecialize}(DUMB_WRAPPER(out, u0, p, first(_tspan), ODE_F_WRAPPER(f)))
+        ODEFunction{iip, FullSpecialize}(DUMB_WRAPPER(p, first(_tspan), ODE_F_WRAPPER(f)))
     else
         _f = ODEFunction{iip, DEFAULT_SPECIALIZATION}(f)
     end

src/scimlfunctions.jl

@@ -2419,23 +2419,19 @@ end
 mutable struct ODE_F_WRAPPER{F}
     const f::F
 end
-mutable struct DUMB_WRAPPER{OUT, U, P, T}
-    out::OUT
-    u::U
+mutable struct DUMB_WRAPPER{P, T}
     p::P
     t::T
     f::ODE_F_WRAPPER
 end
-function (w::ODE_F_WRAPPER)(d::DUMB_WRAPPER)
-    w.f(d.out, d.u, d.p, d.t)
+function (w::ODE_F_WRAPPER)(out, u, d::DUMB_WRAPPER)
+    w.f(out, u, d.p, d.t)
     return nothing
 end
 function (w::DUMB_WRAPPER)(out, u, p, t)
-    w.out = out
-    w.u = u
     w.p = p
     w.t = t
-    w.f(w)
+    w.f(out, u, w)
     return nothing
 end