From 5c4a250bf9ce2197b01b49cd73a0f9b6af1eb92d Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Mon, 17 Jan 2022 02:40:38 -0600 Subject: NCCL: specify cuda_arch (#28433) --- var/spack/repos/builtin/packages/nccl/package.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) (limited to 'var') 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): -- cgit v1.2.3-70-g09d2