fix idx_work sync for parallel runs

ilhamv · Sep 29, 2023 · fcb87e1 · fcb87e1
1 parent 0833787
commit fcb87e1
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/mcdc/loop.py b/mcdc/loop.py
@@ -121,9 +121,10 @@ def loop_source(seed, mcdc):
 
     # Loop over particle sources
     work_start = mcdc["mpi_work_start"]
-    work_end = work_start + mcdc["mpi_work_size"]
-    for idx_work in range(work_start, work_end):
-        seed_work = kernel.split_seed(idx_work, seed)
+    work_size = mcdc['mpi_work_size']
+    work_end = work_start + work_size
+    for idx_work in range(work_size):
+        seed_work = kernel.split_seed(work_start + idx_work, seed)
 
         # Particle tracker
         if mcdc["setting"]["track_particle"]: