Merge pull request #85 from SciML/dae_benchmark

fix benchmarking for DAEs

Author: ChrisRackauckas

Project.toml

@@ -1,7 +1,7 @@
 name = "DiffEqDevTools"
 uuid = "f3b72e0c-5b89-59e1-b016-84e28bfd966d"
 authors = ["Chris Rackauckas <[email protected]>"]
-version = "2.27.2"
+version = "2.27.3"
 
 [deps]
 DiffEqBase = "2b5f629d-d688-5b77-993f-72d75c75574e"

src/benchmark.jl

@@ -204,18 +204,35 @@ function WorkPrecision(prob,alg,abstols,reltols,dts=nothing;
 
       benchmark_f = let dts=dts,_prob=_prob,alg=alg,sol=sol,abstols=abstols,reltols=reltols,kwargs=kwargs
         if dts === nothing
-          () -> @elapsed solve(_prob, alg, sol.u, sol.t, sol.k;
-                               abstol = abstols[i],
-                               reltol = reltols[i],
-                               timeseries_errors = false,
-                               dense_errors = false, kwargs...)
+          if typeof(_prob) <: DAEProblem
+            () -> @elapsed solve(_prob, alg, sol.u, sol.t;
+                                abstol = abstols[i],
+                                reltol = reltols[i],
+                                timeseries_errors = false,
+                                dense_errors = false, kwargs...)
+          else
+            () -> @elapsed solve(_prob, alg, sol.u, sol.t, sol.k;
+                                abstol = abstols[i],
+                                reltol = reltols[i],
+                                timeseries_errors = false,
+                                dense_errors = false, kwargs...)
+          end
         else
-          () -> @elapsed solve(_prob, alg, sol.u, sol.t, sol.k;
-                               abstol = abstols[i],
-                               reltol = reltols[i],
-                               dt = dts[i],
-                               timeseries_errors = false,
-                               dense_errors = false, kwargs...)
+          if typeof(_prob) <: DAEProblem
+            () -> @elapsed solve(_prob, alg, sol.u, sol.t;
+                                abstol = abstols[i],
+                                reltol = reltols[i],
+                                dt = dts[i],
+                                timeseries_errors = false,
+                                dense_errors = false, kwargs...)
+          else
+            () -> @elapsed solve(_prob, alg, sol.u, sol.t, sol.k;
+                                abstol = abstols[i],
+                                reltol = reltols[i],
+                                dt = dts[i],
+                                timeseries_errors = false,
+                                dense_errors = false, kwargs...)
+          end
         end
       end
       benchmark_f() # pre-compile