diff options
| author | Adam J. Stewart <ajstewart426@gmail.com> | 2022-01-17 02:40:38 -0600 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-01-17 09:40:38 +0100 |
| commit | 5c4a250bf9ce2197b01b49cd73a0f9b6af1eb92d (patch) | |
| tree | 14826b33ad5958883b331d0c083772f2821c749c | |
| parent | 5e595586a77dd7a8de25169dfc6cd45cb4cd3a5d (diff) | |
| download | spack-5c4a250bf9ce2197b01b49cd73a0f9b6af1eb92d.tar.gz spack-5c4a250bf9ce2197b01b49cd73a0f9b6af1eb92d.tar.bz2 spack-5c4a250bf9ce2197b01b49cd73a0f9b6af1eb92d.tar.xz spack-5c4a250bf9ce2197b01b49cd73a0f9b6af1eb92d.zip | |
NCCL: specify cuda_arch (#28433)
| -rw-r--r-- | var/spack/repos/builtin/packages/nccl/package.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/var/spack/repos/builtin/packages/nccl/package.py b/var/spack/repos/builtin/packages/nccl/package.py index d7498b8893..c784fd0127 100644 --- a/var/spack/repos/builtin/packages/nccl/package.py +++ b/var/spack/repos/builtin/packages/nccl/package.py @@ -45,10 +45,19 @@ class Nccl(MakefilePackage, CudaPackage): patch('so_reuseport.patch', when='@2.3.7-1:2.4.8-1') conflicts('~cuda', msg='NCCL requires CUDA') + conflicts('cuda_arch=none', + msg='Must specify CUDA compute capabilities of your GPU, see ' + 'https://developer.nvidia.com/cuda-gpus') @property def build_targets(self): - return ['CUDA_HOME={0}'.format(self.spec['cuda'].prefix)] + cuda_arch = self.spec.variants['cuda_arch'].value + cuda_gencode = ' '.join(self.cuda_flags(cuda_arch)) + + return [ + 'CUDA_HOME={0}'.format(self.spec['cuda'].prefix), + 'NVCC_GENCODE={0}'.format(cuda_gencode), + ] @property def install_targets(self): |