some small test time optimizations

Author: ChrisRackauckas

Project.toml

@@ -1,7 +1,7 @@
 name = "DiffEqBase"
 uuid = "2b5f629d-d688-5b77-993f-72d75c75574e"
 authors = ["Chris Rackauckas <[email protected]>"]
-version = "6.34.2"
+version = "6.34.3"
 
 [deps]
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"

src/ensemble/basic_ensemble_solve.jl

@@ -215,7 +215,7 @@ function solve_batch(prob,alg,ensemblealg::EnsembleThreads,II,pmap_batch_size;kw
 
   batch_data = Vector{Core.Compiler.return_type(multithreaded_batch,Tuple{typeof(first(II))})}(undef,length(II))
   let
-    if length(II) == 1
+    if length(II) == 1 || Threads.nthreads() == 1
       for batch_idx in axes(batch_data, 1)
           batch_data[batch_idx] = multithreaded_batch(batch_idx)
       end
@@ -294,7 +294,7 @@ function thread_monte(prob,II,alg,procid;kwargs...)
   batch_data = Vector{Core.Compiler.return_type(multithreaded_batch,Tuple{typeof(first(II))})}(undef,length(II))
 
   let
-    if length(II) == 1
+    if length(II) == 1 || Threads.nthreads() == 1
       for batch_idx in axes(batch_data, 1)
           batch_data[batch_idx] = multithreaded_batch(batch_idx)
       end

src/ensemble/ensemble_solutions.jl

@@ -78,7 +78,8 @@ function calculate_ensemble_errors(u;elapsedTime=0.0,converged=false,
   if weak_dense_errors
     densetimes = collect(range(u[1].t[1], stop=u[1].t[end], length=100))
     u_analytic = [[sol.prob.f(Val{:analytic},sol.prob.u0,sol.prob.p,densetimes[i],sol.W(densetimes[i])[1]) for i in eachindex(densetimes)] for sol in u]
-    dense_weak_errors = [mean([u[j](densetimes)[i] - u_analytic[j][i] for j in 1:length(u)]) for i in eachindex(densetimes)]
+    udense = [u[j](densetimes) for j in 1:length(u)]
+    dense_weak_errors = [mean([udense[j][i] - u_analytic[j][i] for j in 1:length(u)]) for i in eachindex(densetimes)]
     dense_L2_errors = [sqrt.(sum(abs2,err)/length(err)) for err in dense_weak_errors]
     L2_tmp = sqrt(sum(abs2,dense_L2_errors)/length(dense_L2_errors))
     max_tmp = maximum([maximum(abs.(err)) for err in dense_weak_errors])