Skip to content

Add --mpi=pmix to COMMAND srun in Ray Slurm template#439

Closed
athitten wants to merge 2 commits intomainfrom
athitten/raytemplate_mpi
Closed

Add --mpi=pmix to COMMAND srun in Ray Slurm template#439
athitten wants to merge 2 commits intomainfrom
athitten/raytemplate_mpi

Conversation

@athitten
Copy link

@athitten athitten commented Mar 2, 2026

Summary

  • Adds --mpi=pmix to the COMMAND srun in ray.sub.j2 for consistency with all other sruns in the template (head, worker, and overlap sruns already include it via {{ common_srun_args }})

  • This is required to fix MPI initialization failure on clusters where PMIx is required but not automatically configured for srun steps. Adding --mpi=pmix should be harmless for clusters where this was already working

Signed-off-by: Abhishree <abhishreetm@gmail.com>
Signed-off-by: Abhishree <abhishreetm@gmail.com>
@athitten
Copy link
Author

athitten commented Mar 3, 2026

Closing since its fixed in #440

@athitten athitten closed this Mar 3, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant