diff options
Diffstat (limited to 'var/spack/repos/builtin/packages/nvshmem/package.py')
| -rw-r--r-- | var/spack/repos/builtin/packages/nvshmem/package.py | 82 |
1 files changed, 41 insertions, 41 deletions
diff --git a/var/spack/repos/builtin/packages/nvshmem/package.py b/var/spack/repos/builtin/packages/nvshmem/package.py index b2be2d2775..c6214f9d58 100644 --- a/var/spack/repos/builtin/packages/nvshmem/package.py +++ b/var/spack/repos/builtin/packages/nvshmem/package.py @@ -16,58 +16,58 @@ class Nvshmem(MakefilePackage, CudaPackage): homepage = "https://developer.nvidia.com/nvshmem" - maintainers = ['bvanessen'] - - version('2.5.0-19', sha256='dd800b40f1d296e1d3ed2a9885adcfe745c3e57582bc809860e87bd32abcdc60') - version('2.4.1-3', sha256='8b6c0eab321b6352911e470f9e81a777a49e58148ec3728453b9522446dba178') - version('2.2.1-0', sha256='c8efc6cd560e0ed66d5fe4c5837c650247bec7b0dc65b5089deb8ab49658e1c3') - version('2.1.2-0', sha256='367211808df99b4575fb901977d9f4347065c61a26642d65887f24d60342a4ec') - version('2.0.3-0', sha256='20da93e8508511e21aaab1863cb4c372a3bec02307b932144a7d757ea5a1bad2') - - variant('cuda', default=True, description='Build with CUDA') - variant('ucx', default=True, description='Build with UCX support') - variant('nccl', default=True, description='Build with NCCL support') - 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') - conflicts('~cuda') + maintainers = ["bvanessen"] + + version("2.5.0-19", sha256="dd800b40f1d296e1d3ed2a9885adcfe745c3e57582bc809860e87bd32abcdc60") + version("2.4.1-3", sha256="8b6c0eab321b6352911e470f9e81a777a49e58148ec3728453b9522446dba178") + version("2.2.1-0", sha256="c8efc6cd560e0ed66d5fe4c5837c650247bec7b0dc65b5089deb8ab49658e1c3") + version("2.1.2-0", sha256="367211808df99b4575fb901977d9f4347065c61a26642d65887f24d60342a4ec") + version("2.0.3-0", sha256="20da93e8508511e21aaab1863cb4c372a3bec02307b932144a7d757ea5a1bad2") + + variant("cuda", default=True, description="Build with CUDA") + variant("ucx", default=True, description="Build with UCX support") + variant("nccl", default=True, description="Build with NCCL support") + 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") + conflicts("~cuda") def url_for_version(self, version): - ver_str = '{0}'.format(version) - directory = ver_str.split('-')[0] + ver_str = "{0}".format(version) + directory = ver_str.split("-")[0] url_fmt = "https://developer.download.nvidia.com/compute/redist/nvshmem/{0}/source/nvshmem_src_{1}.txz" return url_fmt.format(directory, version) - depends_on('mpi', when='+mpi') - depends_on('ucx', when='+ucx') - depends_on('gdrcopy', when='+gdrcopy') - depends_on('nccl', when='+nccl') + depends_on("mpi", when="+mpi") + depends_on("ucx", when="+ucx") + depends_on("gdrcopy", when="+gdrcopy") + depends_on("nccl", when="+nccl") def setup_build_environment(self, env): - env.set('CUDA_HOME', self.spec['cuda'].prefix) - env.set('NVSHMEM_PREFIX', self.prefix) + env.set("CUDA_HOME", self.spec["cuda"].prefix) + env.set("NVSHMEM_PREFIX", self.prefix) - if '+ucx' in self.spec: - env.set('NVSHMEM_UCX_SUPPORT', '1') - env.set('UCX_HOME', self.spec['ucx'].prefix) + if "+ucx" in self.spec: + env.set("NVSHMEM_UCX_SUPPORT", "1") + env.set("UCX_HOME", self.spec["ucx"].prefix) - if '+gdrcopy' in self.spec: - env.set('NVSHMEM_USE_GDRCOPY', '1') - env.set('GDRCOPY_HOME', self.spec['gdrcopy'].prefix) + if "+gdrcopy" in self.spec: + env.set("NVSHMEM_USE_GDRCOPY", "1") + env.set("GDRCOPY_HOME", self.spec["gdrcopy"].prefix) - if '+nccl' in self.spec: - env.set('NVSHMEM_USE_NCCL', '1') - env.set('NCCL_HOME', self.spec['nccl'].prefix) + if "+nccl" in self.spec: + env.set("NVSHMEM_USE_NCCL", "1") + env.set("NCCL_HOME", self.spec["nccl"].prefix) - if '+mpi' in self.spec: - env.set('NVSHMEM_MPI_SUPPORT', '1') - env.set('MPI_HOME', self.spec['mpi'].prefix) + if "+mpi" in self.spec: + env.set("NVSHMEM_MPI_SUPPORT", "1") + env.set("MPI_HOME", self.spec["mpi"].prefix) - if self.spec.satisfies('^spectrum-mpi') or self.spec.satisfies('^openmpi'): - env.set('NVSHMEM_MPI_IS_OMPI', '1') + if self.spec.satisfies("^spectrum-mpi") or self.spec.satisfies("^openmpi"): + env.set("NVSHMEM_MPI_IS_OMPI", "1") else: - env.set('NVSHMEM_MPI_IS_OMPI', '0') + env.set("NVSHMEM_MPI_IS_OMPI", "0") - if '+shmem' in self.spec: - env.set('NVSHMEM_SHMEM_SUPPORT', '1') - env.set('SHMEM_HOME', self.spec['mpi'].prefix) + if "+shmem" in self.spec: + env.set("NVSHMEM_SHMEM_SUPPORT", "1") + env.set("SHMEM_HOME", self.spec["mpi"].prefix) |