diff options
| author | Adam J. Stewart <ajstewart426@gmail.com> | 2020-06-24 21:59:32 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-06-24 21:59:32 -0500 |
| commit | 791267c3dcd539f604a86f381582db5e28439b9f (patch) | |
| tree | 87b4cb74f756643a3ece66057a18829dd7ba4c23 | |
| parent | cd2e945e4e2626d71b3f6d6b54924b3e6de5cae7 (diff) | |
| download | spack-791267c3dcd539f604a86f381582db5e28439b9f.tar.gz spack-791267c3dcd539f604a86f381582db5e28439b9f.tar.bz2 spack-791267c3dcd539f604a86f381582db5e28439b9f.tar.xz spack-791267c3dcd539f604a86f381582db5e28439b9f.zip | |
NCCL: add latest version (#17234)
| -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)] |