Skip to content

Commit

Permalink
Change map for a for loop to work around
Browse files Browse the repository at this point in the history
JuliaLang/julia#35800

which can cause very slow compile times in Julia 1.6.
  • Loading branch information
andreasnoack committed Mar 31, 2021
1 parent 3150439 commit 87beef2
Showing 1 changed file with 8 additions and 4 deletions.
12 changes: 8 additions & 4 deletions src/ensemble/basic_ensemble_solve.jl
Original file line number Diff line number Diff line change
Expand Up @@ -182,11 +182,15 @@ function solve_batch(prob,alg,ensemblealg::EnsembleDistributed,II,pmap_batch_siz
tighten_container_eltype(batch_data)
end

function solve_batch(prob,alg,::EnsembleSerial,II,pmap_batch_size;kwargs...)
batch_data = map(II) do i
batch_func(i,prob,alg;kwargs...)
function solve_batch(prob, alg, ::EnsembleSerial, II, pmap_batch_size; kwargs...)
if isempty(II)
throw(ArgumentError("number of trajectories must be positive"))
end
tighten_container_eltype(batch_data)
batch_data = [batch_func(first(II), prob, alg; kwargs...)]
for i in 2:length(II)
@inbounds push!(batch_data, batch_func(II[i], prob, alg; kwargs...))
end
return tighten_container_eltype(batch_data)
end

function solve_batch(prob,alg,ensemblealg::EnsembleThreads,II,pmap_batch_size;kwargs...)
Expand Down

0 comments on commit 87beef2

Please sign in to comment.