mvapich: add v4.0 and update default pmi version (#49399)

Co-authored-by: Matt Lieber <lieber.31@osu.edu>
This commit is contained in:
MatthewLieber 2025-03-17 04:38:19 -04:00 committed by GitHub
parent 38e9043b9e
commit 78abe968a0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -26,6 +26,7 @@ class Mvapich(MpichEnvironmentModifications, AutotoolsPackage):
license("Unlicense") license("Unlicense")
# Prefer the latest stable release # Prefer the latest stable release
version("4.0", sha256="c532f7bdd5cca71f78c12e0885c492f6e276e283711806c84d0b0f80bb3e3b74")
version("3.0", sha256="ee076c4e672d18d6bf8dd2250e4a91fa96aac1db2c788e4572b5513d86936efb") version("3.0", sha256="ee076c4e672d18d6bf8dd2250e4a91fa96aac1db2c788e4572b5513d86936efb")
depends_on("c", type="build") depends_on("c", type="build")
@ -67,8 +68,8 @@ class Mvapich(MpichEnvironmentModifications, AutotoolsPackage):
variant( variant(
"pmi_version", "pmi_version",
description="Which pmi version to be used. If using pmi2 add it to your CFLAGS", description="Which pmi version to be used. If using pmi2 add it to your CFLAGS",
default="simple", default="none",
values=("simple", "pmi2", "pmix"), values=("none", "pmi1", "pmi2", "pmix"),
multi=False, multi=False,
) )
@ -163,7 +164,6 @@ def process_manager_options(self):
if "process_managers=slurm" in spec: if "process_managers=slurm" in spec:
opts = [ opts = [
"--with-pm=slurm", "--with-pm=slurm",
"--with-pmi=simple",
"--with-slurm={0}".format(spec["slurm"].prefix), "--with-slurm={0}".format(spec["slurm"].prefix),
"CFLAGS=-I{0}/include/slurm".format(spec["slurm"].prefix), "CFLAGS=-I{0}/include/slurm".format(spec["slurm"].prefix),
] ]
@ -231,7 +231,8 @@ def configure_args(self):
] ]
args.extend(self.enable_or_disable("alloca")) args.extend(self.enable_or_disable("alloca"))
args.append("--with-pmi=" + spec.variants["pmi_version"].value) if not spec.satisfies("pmi_version=none"):
args.append("--with-pmi=" + spec.variants["pmi_version"].value)
if "pmi_version=pmix" in spec: if "pmi_version=pmix" in spec:
args.append("--with-pmix={0}".format(spec["pmix"].prefix)) args.append("--with-pmix={0}".format(spec["pmix"].prefix))