hyperion-rt / hyperion

Hyperion Radiative Transfer Code
http://www.hyperion-rt.org
BSD 2-Clause "Simplified" License
52 stars 26 forks source link

slurm srun usage #202

Open dnarayanan opened 6 years ago

dnarayanan commented 6 years ago

I'd like to edit scripts/hyperion to include something like:

if os.getenv("SLURM_SUBMIT_HOST") is None:
     mpi_command = "mpiexec"                   # SLURM?  No.
  else:
     mpi_command = "srun --mpi=pmix"           # SLURM?  Yes.

so that if the scheduling environment is SLURM, MPI will be run with srun instead of mpirun which is preferred (and, at least in our clusters case, required). I forked hyperion to do this and create a PR, but I can't figure out how to edit the code that goes to conda -- so more than an issue, this is a question as to how I can do that!