diff --git a/src/scimlfunctions.jl b/src/scimlfunctions.jl index 00d888fbb..c52bf700b 100644 --- a/src/scimlfunctions.jl +++ b/src/scimlfunctions.jl @@ -2644,6 +2644,26 @@ function Base.getproperty(f::DynamicalDDEFunction, name::Symbol) end (f::SDEFunction)(args...) = f.f(args...) + +@static if isdefined(SciMLOperators, :isv1) + function (f::SDEFunction)(du, u, p, t) + if f.f isa AbstractSciMLOperator + f.f(du, u, u, p, t) + else + f.f(du, u, p, t) + end + end + + function (f::SDEFunction)(u, p, t) + if f.f isa AbstractSciMLOperator + f.f(u, u, p, t) + else + f.f(u, p, t) + end + end +end + + (f::SDDEFunction)(args...) = f.f(args...) (f::SplitSDEFunction)(u, p, t) = f.f1(u, p, t) + f.f2(u, p, t)