zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added 2.6.0 version to NVSHMEM package. Added new variant for GPU (#32622)

Browse Source 
initiated support experimental feature.
This commit is contained in:
Brian Van Essen 2022-09-13 10:14:04 -04:00 committed by GitHub
parent 62e788fb89
commit 2d895a9ec3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
var/spack/repos/builtin/packages/nvshmem/package.py
View File

@ -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 @@ def setup_build_environment(self, env):
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")