Merge pull request #1035 from SciML/ChrisRackauckas-patch-2

Make SciMLOperator dispatch safer

Author: ChrisRackauckas

Project.toml

@@ -85,7 +85,7 @@ RecipesBase = "1.3.4"
 RecursiveArrayTools = "3.27.2"
 Reexport = "1"
 RuntimeGeneratedFunctions = "0.5.12"
-SciMLOperators = "0.4, 1"
+SciMLOperators = "0.4, 1.3"
 SciMLStructures = "1.1"
 StableRNGs = "1.0"
 StaticArrays = "1.7"

src/scimlfunctions.jl

@@ -2571,19 +2571,23 @@ end
 ######### Backwards Compatibility Overloads
 
 (f::ODEFunction)(args...) = f.f(args...)
-function (f::ODEFunction)(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::ODEFunction)(u, p, t) 
-    if f.f isa AbstractSciMLOperator
-        f.f(u, u, p, t)
-    else
-        f.f(u, p, t)
-    end
+
+@static if isdefined(SciMLOperators, :isv1)
+  function (f::ODEFunction)(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::ODEFunction)(u, p, t) 
+      if f.f isa AbstractSciMLOperator
+          f.f(u, u, p, t)
+      else
+          f.f(u, p, t)
+      end
+  end
 end
 
 (f::NonlinearFunction)(args...) = f.f(args...)