zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add additional comments about logic to choose a scheduler to run MPI

Browse Source
This commit is contained in:
Tara Drwenski 2025-03-24 16:08:50 -06:00
parent 51930949c4
commit dcf19395fa
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lib/spack/spack/build_systems/cached_cmake.py
View File

@ -189,14 +189,18 @@ class Scheduler(enum.Enum):
def get_scheduler(self) -> Optional[Scheduler]: def get_scheduler(self) -> Optional[Scheduler]:
spec = self.pkg.spec spec = self.pkg.spec
# Check for Spectrum-mpi, which always uses LSF or LSF MPI variant
if spec.satisfies("^spectrum-mpi") or spec["mpi"].satisfies("schedulers=lsf"): if spec.satisfies("^spectrum-mpi") or spec["mpi"].satisfies("schedulers=lsf"):
return self.Scheduler.LSF return self.Scheduler.LSF
# Check for Slurm MPI variants
slurm_checks = ["+slurm", "schedulers=slurm", "process_managers=slurm"] slurm_checks = ["+slurm", "schedulers=slurm", "process_managers=slurm"]
if any(spec["mpi"].satisfies(variant) for variant in slurm_checks): if any(spec["mpi"].satisfies(variant) for variant in slurm_checks):
return self.Scheduler.SLURM return self.Scheduler.SLURM
# TODO improve this when MPI implementations support flux # TODO improve this when MPI implementations support flux
# Do this check last to avoid using a flux wrapper present next to Slurm/ LSF schedulers
if which_string("flux") is not None: if which_string("flux") is not None:
return self.Scheduler.FLUX return self.Scheduler.FLUX