From 34aba94148ef9838200dafa541d80b5fb3a3102c Mon Sep 17 00:00:00 2001 From: Carlos Bederián <4043375+zzzoom@users.noreply.github.com> Date: Fri, 16 Feb 2024 18:37:00 -0300 Subject: openmpi: add ucc to fabrics (#41889) --- var/spack/repos/builtin/packages/openmpi/package.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/var/spack/repos/builtin/packages/openmpi/package.py b/var/spack/repos/builtin/packages/openmpi/package.py index a05b6ed11e..b076fec15e 100644 --- a/var/spack/repos/builtin/packages/openmpi/package.py +++ b/var/spack/repos/builtin/packages/openmpi/package.py @@ -443,6 +443,7 @@ class Openmpi(AutotoolsPackage, CudaPackage): "ofi", "fca", "hcoll", + "ucc", "xpmem", "cma", "knem", @@ -573,6 +574,7 @@ class Openmpi(AutotoolsPackage, CudaPackage): depends_on("libfabric", when="fabrics=ofi") depends_on("fca", when="fabrics=fca") depends_on("hcoll", when="fabrics=hcoll") + depends_on("ucc", when="fabrics=ucc") depends_on("xpmem", when="fabrics=xpmem") depends_on("knem", when="fabrics=knem") @@ -614,6 +616,8 @@ class Openmpi(AutotoolsPackage, CudaPackage): conflicts("fabrics=fca", when="@:1.4,5:") # hcoll support was added in 1.7.3: conflicts("fabrics=hcoll", when="@:1.7.2") + # ucc support was added in 4.1.4: + conflicts("fabrics=ucc", when="@:4.1.3") # xpmem support was added in 1.7 conflicts("fabrics=xpmem", when="@:1.6") # cma support was added in 1.7 @@ -907,6 +911,11 @@ class Openmpi(AutotoolsPackage, CudaPackage): return "--without-hcoll" return "--with-hcoll={0}".format(self.spec["hcoll"].prefix) + def with_or_without_ucc(self, activated): + if not activated: + return "--without-ucc" + return "--with-ucc={0}".format(self.spec["ucc"].prefix) + def with_or_without_xpmem(self, activated): if not activated: return "--without-xpmem" -- cgit v1.2.3-70-g09d2