diff options
| -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") |