summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorBrian Van Essen <vanessen1@llnl.gov>2022-09-13 10:14:04 -0400
committerGitHub <noreply@github.com>2022-09-13 08:14:04 -0600
commit2d895a9ec3a73c6b749741375c7cb67d13a181cd (patch)
tree5e251f2513940b914c730dd631e081df64af1738
parent62e788fb894bfe4caa5a925165965a58defa1712 (diff)
downloadspack-2d895a9ec3a73c6b749741375c7cb67d13a181cd.tar.gz
spack-2d895a9ec3a73c6b749741375c7cb67d13a181cd.tar.bz2
spack-2d895a9ec3a73c6b749741375c7cb67d13a181cd.tar.xz
spack-2d895a9ec3a73c6b749741375c7cb67d13a181cd.zip
Added 2.6.0 version to NVSHMEM package. Added new variant for GPU (#32622)
initiated support experimental feature.
-rw-r--r--var/spack/repos/builtin/packages/nvshmem/package.py10
1 files changed, 10 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/nvshmem/package.py b/var/spack/repos/builtin/packages/nvshmem/package.py
index c6214f9d58..d007c6c611 100644
--- a/var/spack/repos/builtin/packages/nvshmem/package.py
+++ b/var/spack/repos/builtin/packages/nvshmem/package.py
@@ -18,6 +18,7 @@ class Nvshmem(MakefilePackage, CudaPackage):
maintainers = ["bvanessen"]
+ version("2.6.0-1", sha256="fc0e8de61b034f3a079dc231b1d0955e665a9f57b5013ee98b6743647bd60417")
version("2.5.0-19", sha256="dd800b40f1d296e1d3ed2a9885adcfe745c3e57582bc809860e87bd32abcdc60")
version("2.4.1-3", sha256="8b6c0eab321b6352911e470f9e81a777a49e58148ec3728453b9522446dba178")
version("2.2.1-0", sha256="c8efc6cd560e0ed66d5fe4c5837c650247bec7b0dc65b5089deb8ab49658e1c3")
@@ -30,6 +31,12 @@ class Nvshmem(MakefilePackage, CudaPackage):
variant("gdrcopy", default=True, description="Build with gdrcopy support")
variant("mpi", default=True, description="Build with MPI support")
variant("shmem", default=False, description="Build with shmem support")
+ variant(
+ "gpu_initiated_support",
+ default=False,
+ when="@2.6:",
+ description="Build with support for GPU initiated communication",
+ )
conflicts("~cuda")
def url_for_version(self, version):
@@ -71,3 +78,6 @@ class Nvshmem(MakefilePackage, CudaPackage):
if "+shmem" in self.spec:
env.set("NVSHMEM_SHMEM_SUPPORT", "1")
env.set("SHMEM_HOME", self.spec["mpi"].prefix)
+
+ if "+gpu_initiated_support" in self.spec:
+ env.set("NVSHMEM_GPUINITIATED_SUPPORT", "1")