From f31a99f18828f2725b3c3fbed1191562d99c56d2 Mon Sep 17 00:00:00 2001 From: MatthewLieber <77356607+MatthewLieber@users.noreply.github.com> Date: Sat, 28 Sep 2024 09:17:10 -0400 Subject: [PATCH] mvapich: Add pmix variant (#45531) --- var/spack/repos/builtin/packages/mvapich/package.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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(