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

mvapich: Add pmix variant (#45531)

Browse Source
This commit is contained in:
MatthewLieber 2024-09-28 09:17:10 -04:00 committed by GitHub
parent 6c4abef75c
commit f31a99f188
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
var/spack/repos/builtin/packages/mvapich/package.py
View File

@ -69,7 +69,7 @@ class Mvapich(AutotoolsPackage):
"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="simple",
values=("simple", "pmi2"), values=("simple", "pmi2", "pmix"),
multi=False, multi=False,
) )
@ -113,6 +113,7 @@ class Mvapich(AutotoolsPackage):
depends_on("libfabric", when="netmod=ofi") depends_on("libfabric", when="netmod=ofi")
depends_on("slurm", when="process_managers=slurm") depends_on("slurm", when="process_managers=slurm")
depends_on("ucx", when="netmod=ucx") depends_on("ucx", when="netmod=ucx")
depends_on("pmix", when="pmi_version=pmix")
with when("process_managers=slurm"): with when("process_managers=slurm"):
conflicts("pmi_version=pmi2") conflicts("pmi_version=pmi2")
@ -265,6 +266,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) args.append("--with-pmi=" + spec.variants["pmi_version"].value)
if "pmi_version=pmix" in spec:
args.append("--with-pmix={0}".format(spec["pmix"].prefix))
if "+debug" in self.spec: if "+debug" in self.spec:
args.extend( args.extend(