Skip to content

Commit 9811e5a

Browse files
Merge pull request #161 from JuliaDiffEq/myb/Wfact
Fix Wfact
2 parents a96227d + 336c737 commit 9811e5a

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

src/systems/diffeqs/diffeqsystem.jl

+9-9
Original file line numberDiff line numberDiff line change
@@ -232,14 +232,14 @@ function DiffEqBase.ODEFunction{iip}(sys::ODESystem, dvs, ps,
232232

233233
if Wfact
234234
_Wfact,_Wfact_t = eval.(generate_factorized_W(sys, dvs, ps))
235-
Wfact_f_safe(u,p,t) = ModelingToolkit.fast_invokelatest(_Wfact,Matrix{eltype(u)},u,p,t)
236-
Wfact_f_safe(J,u,p,t) = ModelingToolkit.fast_invokelatest(_Wfact,Nothing,J,u,p,t)
237-
Wfact_f_t_safe(u,p,t) = ModelingToolkit.fast_invokelatest(_Wfact,Matrix{eltype(u)},u,p,t)
238-
Wfact_f_t_safe(J,u,p,t) = ModelingToolkit.fast_invokelatest(_Wfact,Nothing,J,u,p,t)
239-
Wfact_f(u,p,t) = _Wfact(u,p,t)
240-
Wfact_f(J,u,p,t) = _Wfact(J,u,p,t)
241-
Wfact_f_t(u,p,t) = _Wfact_t(u,p,t)
242-
Wfact_f_t(J,u,p,t) = _Wfact_t(J,u,p,t)
235+
Wfact_f_safe(u,p,gam,t) = ModelingToolkit.fast_invokelatest(_Wfact,Matrix{eltype(u)},u,p,gam,t)
236+
Wfact_f_safe(J,u,p,gam,t) = ModelingToolkit.fast_invokelatest(_Wfact,Nothing,J,u,p,gam,t)
237+
Wfact_f_t_safe(u,p,gam,t) = ModelingToolkit.fast_invokelatest(_Wfact,Matrix{eltype(u)},u,p,gam,t)
238+
Wfact_f_t_safe(J,u,p,gam,t) = ModelingToolkit.fast_invokelatest(_Wfact,Nothing,J,u,p,gam,t)
239+
Wfact_f(u,p,gam,t) = _Wfact(u,p,gam,t)
240+
Wfact_f(J,u,p,gam,t) = _Wfact(J,u,p,gam,t)
241+
Wfact_f_t(u,p,gam,t) = _Wfact_t(u,p,gam,t)
242+
Wfact_f_t(J,u,p,gam,t) = _Wfact_t(J,u,p,gam,t)
243243
else
244244
Wfact_f_safe = nothing
245245
Wfact_f_t_safe = nothing
@@ -254,7 +254,7 @@ function DiffEqBase.ODEFunction{iip}(sys::ODESystem, dvs, ps,
254254
else
255255
ODEFunction{iip}(out_f,jac=jac_f,
256256
Wfact = Wfact_f,
257-
Wfact_t = Wfact_t_f)
257+
Wfact_t = Wfact_f_t)
258258
end
259259
end
260260

0 commit comments

Comments
 (0)