Skip to content

Commit c9df021

Browse files
committed
Fix MPI write
1 parent 77891c6 commit c9df021

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

bin/run_vega_mc_mpi.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,4 +76,7 @@ def print_func(message):
7676
vega.analysis.run_monte_carlo(fiducial_model, num_mocks=num_local_mc, seed=local_seed)
7777

7878
# Write output
79-
vega.output.write_monte_carlo()
79+
if num_cpus > 1:
80+
vega.output.write_monte_carlo(cpu_rank)
81+
else:
82+
vega.output.write_monte_carlo()

0 commit comments

Comments
 (0)