Skip to content

Commit 3ba02de

Browse files
Apply suggestions from code review
1 parent b0eb83c commit 3ba02de

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/problems/ode_problems.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -474,7 +474,7 @@ end
474474
function SplitODEProblem{iip}(f::SplitFunction, u0, tspan, p = NullParameters();
475475
kwargs...) where {iip}
476476
if f._func_cache === nothing && iip
477-
_func_cache = PreallocationTools.DiffCache(u0)
477+
_func_cache = DiffCache(u0)
478478
f = remake(f; _func_cache)
479479
end
480480
ODEProblem(f, u0, tspan, p, SplitODEProblem{iip}(); kwargs...)

src/scimlfunctions.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2618,7 +2618,7 @@ end
26182618

26192619
(f::SplitFunction)(u, p, t) = f.f1(u, p, t) + f.f2(u, p, t)
26202620
function (f::SplitFunction)(du, u, p, t)
2621-
tmp = PreallocationTools.get_tmp(f._func_cache, u)
2621+
tmp = get_tmp(f._func_cache, u)
26222622
f.f1(tmp, u, p, t)
26232623
f.f2(du, u, p, t)
26242624
du .+= tmp
@@ -2668,7 +2668,7 @@ end
26682668
(f::SDDEFunction)(args...) = f.f(args...)
26692669
(f::SplitSDEFunction)(u, p, t) = f.f1(u, p, t) + f.f2(u, p, t)
26702670
function (f::SplitSDEFunction)(du, u, p, t)
2671-
tmp = PreallocationTools.get_tmp(f._func_cache)
2671+
tmp = get_tmp(f._func_cache)
26722672
f.f1(tmp, u, p, t)
26732673
f.f2(du, u, p, t)
26742674
du .+= tmp

0 commit comments

Comments
 (0)