From 5cfb220160231631d2fa2ae780eeec9a4af2a3cc Mon Sep 17 00:00:00 2001 From: Chris Rackauckas Date: Wed, 14 Apr 2021 16:10:23 -0400 Subject: [PATCH 1/2] type-stable responsible map --- src/ensemble/basic_ensemble_solve.jl | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/ensemble/basic_ensemble_solve.jl b/src/ensemble/basic_ensemble_solve.jl index cd41e874b..d75a1b872 100644 --- a/src/ensemble/basic_ensemble_solve.jl +++ b/src/ensemble/basic_ensemble_solve.jl @@ -183,12 +183,12 @@ function solve_batch(prob,alg,ensemblealg::EnsembleDistributed,II,pmap_batch_siz end function responsible_map(f,II...) - batch_data = [] + batch_data = Vector{Core.Compiler.return_type(f,Tuple{typeof.(getindex.(II,1))...})}(undef,length(II[1])) sizehint!(batch_data,length(II[1])) for i in 1:length(II[1]) - @inbounds push!(batch_data, f(getindex.(II,i)...)) + batch_data[i] = f(getindex.(II,i)...)) end - identity.(batch_data) + batch_data end function SciMLBase.solve_batch(prob,alg,::EnsembleSerial,II,pmap_batch_size;kwargs...) From ec9b4f4370cfb4d65cfb9f5526fbc0b249a96c11 Mon Sep 17 00:00:00 2001 From: Chris Rackauckas Date: Wed, 14 Apr 2021 16:16:34 -0400 Subject: [PATCH 2/2] Typo --- src/ensemble/basic_ensemble_solve.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/ensemble/basic_ensemble_solve.jl b/src/ensemble/basic_ensemble_solve.jl index d75a1b872..a350f27fb 100644 --- a/src/ensemble/basic_ensemble_solve.jl +++ b/src/ensemble/basic_ensemble_solve.jl @@ -186,7 +186,7 @@ function responsible_map(f,II...) batch_data = Vector{Core.Compiler.return_type(f,Tuple{typeof.(getindex.(II,1))...})}(undef,length(II[1])) sizehint!(batch_data,length(II[1])) for i in 1:length(II[1]) - batch_data[i] = f(getindex.(II,i)...)) + batch_data[i] = f(getindex.(II,i)...) end batch_data end