[mpiexecjl] Return exit code of the root rank

JuliaParallel · Jun 4, 2024 · 3dc1925 · 3dc1925
1 parent 7b97da7
commit 3dc1925
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/bin/mpiexecjl b/bin/mpiexecjl
@@ -61,7 +61,10 @@ fi
 SCRIPT='
 using MPI
 ENV["JULIA_PROJECT"] = dirname(Base.active_project())
-mpiexec(exe -> run(`$exe $ARGS`))
+mpiexec()
+p = mpiexec(exe -> run(pipeline(`$(exe) $(ARGS)`; stdout, stderr); wait=false))
+wait(p)
+exit(p.exitcode)
 '
 
 if [ -n "${PROJECT_ARG}" ]; then