diff options
| author | Brian Van Essen <vanessen1@llnl.gov> | 2022-09-13 10:14:04 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-09-13 08:14:04 -0600 |
| commit | 2d895a9ec3a73c6b749741375c7cb67d13a181cd (patch) | |
| tree | 5e251f2513940b914c730dd631e081df64af1738 | |
| parent | 62e788fb894bfe4caa5a925165965a58defa1712 (diff) | |
| download | spack-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.py | 10 |
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") |