Multiprocessing issue fixed (#43)

* Multiprocessing fixed
ODINN-SciML · Dec 14, 2023 · 54476a4 · 54476a4
1 parent 7281def
commit 54476a4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/setup/config.jl b/src/setup/config.jl
@@ -40,8 +40,9 @@ function clean()
     exit()
  end
 
- function enable_multiprocessing(procs::Int)
-    if procs > 0
+ function enable_multiprocessing(params::Sleipnir.Parameters)
+    procs::Int = params.simulation.workers
+    if procs > 0 && params.simulation.multiprocessing
         if nprocs() < procs
             @eval begin
             addprocs($procs - nprocs(); exeflags="--project")

diff --git a/src/simulations/predictions/prediction_utils.jl b/src/simulations/predictions/prediction_utils.jl
@@ -8,7 +8,7 @@ In-place run of the model.
 """
 function run!(simulation::Prediction)
 
-    enable_multiprocessing(simulation.parameters.simulation.workers)
+    enable_multiprocessing(simulation.parameters)
 
     println("Running forward in-place PDE ice flow model...\n")
     results_list = @showprogress pmap((glacier_idx) -> batch_iceflow_PDE!(glacier_idx, simulation), 1:length(simulation.glaciers))