diff options
Diffstat (limited to 'var')
| -rw-r--r-- | var/spack/repos/builtin/packages/nccl/package.py | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/var/spack/repos/builtin/packages/nccl/package.py b/var/spack/repos/builtin/packages/nccl/package.py index 74a45ba13a..d1f4de962d 100644 --- a/var/spack/repos/builtin/packages/nccl/package.py +++ b/var/spack/repos/builtin/packages/nccl/package.py @@ -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)] |