Save dts and sol.stats into WorkPrecision

Author: nathanaelbosch

src/benchmark.jl

@@ -161,6 +161,8 @@ mutable struct WorkPrecision
     reltols::Any
     errors::Any
     times::Any
+    dts::Any
+    stats::Any
     name::Any
     N::Int
 end
@@ -183,6 +185,7 @@ function WorkPrecision(prob, alg, abstols, reltols, dts = nothing;
     N = length(abstols)
     errors = Vector{Float64}(undef, N)
     times = Vector{Float64}(undef, N)
+    stats = Vector{SciMLBase.DEStats}(undef, N)
     if name === nothing
         name = "WP-Alg"
     end
@@ -210,6 +213,8 @@ function WorkPrecision(prob, alg, abstols, reltols, dts = nothing;
                             dense_errors = dense_errors)
             end
 
+            stats[i] = sol.stats
+
             if haskey(kwargs, :prob_choice)
                 cur_appxsol = appxsol[kwargs[:prob_choice]]
             elseif prob isa AbstractArray
@@ -270,7 +275,7 @@ function WorkPrecision(prob, alg, abstols, reltols, dts = nothing;
             end
         end
     end
-    return WorkPrecision(prob, abstols, reltols, errors, times, name, N)
+    return WorkPrecision(prob, abstols, reltols, errors, times, dts, stats, name, N)
 end
 
 # Work precision information for a BVP