diff --git a/var/spack/repos/builtin/packages/mvapich/package.py b/var/spack/repos/builtin/packages/mvapich/package.py index 37221fec277..de5d5e64942 100644 --- a/var/spack/repos/builtin/packages/mvapich/package.py +++ b/var/spack/repos/builtin/packages/mvapich/package.py @@ -69,7 +69,7 @@ class Mvapich(AutotoolsPackage): "pmi_version", description="Which pmi version to be used. If using pmi2 add it to your CFLAGS", default="simple", - values=("simple", "pmi2"), + values=("simple", "pmi2", "pmix"), multi=False, ) @@ -113,6 +113,7 @@ class Mvapich(AutotoolsPackage): depends_on("libfabric", when="netmod=ofi") depends_on("slurm", when="process_managers=slurm") depends_on("ucx", when="netmod=ucx") + depends_on("pmix", when="pmi_version=pmix") with when("process_managers=slurm"): conflicts("pmi_version=pmi2") @@ -265,6 +266,8 @@ def configure_args(self): args.extend(self.enable_or_disable("alloca")) 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: args.extend(