Skip to content

Commit 4a1b176

Browse files
committed
add hook for MPITRAMPOLINE_MPIEXEC
1 parent 44a9dc7 commit 4a1b176

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/MPI.jl

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -124,6 +124,10 @@ function __init__()
124124
ENV["UCX_ERROR_SIGNALS"] = "SIGILL,SIGBUS,SIGFPE"
125125
end
126126

127+
if MPIPreferences.binary == "MPItrampoline_jll" && !haskey(ENV, "MPITRAMPOLINE_MPIEXEC")
128+
ENV["MPITRAMPOLINE_MPIEXEC"] = MPItrampoline_jll.mpich_mpiexec_path
129+
end
130+
127131
run_load_time_hooks()
128132

129133
@require CUDA="052768ef-5323-5732-b1bb-66c8b64840ba" include("cuda.jl")

0 commit comments

Comments
 (0)