Skip to content

Commit 3345d4d

Browse files
fix derivative wrapper location
1 parent e515182 commit 3345d4d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/caches/sdirk_caches.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ end
6868

6969
function alg_cache(alg::ImplicitEM,prob,u,ΔW,ΔZ,rate_prototype,noise_rate_prototype,
7070
uEltypeNoUnits,tTypeNoUnits,uprev,f,t,::Type{Val{false}})
71-
uf = UDerivativeWrapper(f,t)
71+
uf = DiffEqDiffTools.UDerivativeWrapper(f,t)
7272
ηold = one(uEltypeNoUnits)
7373

7474
if alg.κ != nothing
@@ -152,7 +152,7 @@ end
152152

153153
function alg_cache(alg::ImplicitEulerHeun,prob,u,ΔW,ΔZ,rate_prototype,noise_rate_prototype,
154154
uEltypeNoUnits,tTypeNoUnits,uprev,f,t,::Type{Val{false}})
155-
uf = UDerivativeWrapper(f,t)
155+
uf = DiffEqDiffTools.UDerivativeWrapper(f,t)
156156
ηold = one(uEltypeNoUnits)
157157

158158
if alg.κ != nothing
@@ -238,7 +238,7 @@ end
238238

239239
function alg_cache(alg::ImplicitRKMil,prob,u,ΔW,ΔZ,rate_prototype,noise_rate_prototype,
240240
uEltypeNoUnits,tTypeNoUnits,uprev,f,t,::Type{Val{false}})
241-
uf = UDerivativeWrapper(f,t)
241+
uf = DiffEqDiffTools.UDerivativeWrapper(f,t)
242242
ηold = one(uEltypeNoUnits)
243243

244244
if alg.κ != nothing

0 commit comments

Comments
 (0)