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

NCCL: add latest version (#17234)

Browse Source
This commit is contained in:
Adam J. Stewart 2020-06-24 21:59:32 -05:00 committed by GitHub
parent cd2e945e4e
commit 791267c3dc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -6,14 +6,15 @@
from spack import *
class Nccl(MakefilePackage):
class Nccl(MakefilePackage, CudaPackage):
"""Optimized primitives for collective multi-GPU communication."""
homepage = "https://github.com/NVIDIA/nccl"
url = "https://github.com/NVIDIA/nccl/archive/v2.6.4-1.tar.gz"
url = "https://github.com/NVIDIA/nccl/archive/v2.7.3-1.tar.gz"
maintainers = ['adamjstewart']
version('2.7.3-1', sha256='dc7b8794373306e323363314c3327796e416f745e8003490fc1407a22dd7acd6')
version('2.6.4-1', sha256='ed8c9dfd40e013003923ae006787b1a30d3cb363b47d2e4307eaa2624ebba2ba')
version('2.5.7-1', sha256='781a6bb2278566be4abbdf22b2fa19afc7306cff4b312c82bd782979b368014e')
version('2.5.6-2', sha256='8a30e0b4813a825592872fcbeeede22a659e2c399074dcce02960591dc81387d')
@ -26,12 +27,15 @@ class Nccl(MakefilePackage):
version('1.3.4-1', sha256='11e4eb44555bb28b9cbad973dacb4640b82710c9769e719afc2013b63ffaf884')
version('1.3.0-1', sha256='53f36151061907bdcafad1c26c1d9370a0a8400f561a83704a5138213ba51003')
depends_on('cuda')
variant('cuda', default=True, description='Build with CUDA')
depends_on('rdma-core', when='@2.3.5-5:')
# https://github.com/NVIDIA/nccl/issues/244
patch('so_reuseport.patch', when='@2.3.7-1:2.4.8-1')
conflicts('~cuda', msg='NCCL requires CUDA')
@property
def build_targets(self):
return ['CUDA_HOME={0}'.format(self.spec['cuda'].prefix)]