From f2f58b70b1153621d9b42fa2f9045d81843692a7 Mon Sep 17 00:00:00 2001 From: Andrew W Elble Date: Fri, 26 Feb 2021 18:38:20 -0500 Subject: nccl-tests: respect cuda_arch variant during build (#21995) --- var/spack/repos/builtin/packages/nccl-tests/package.py | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'var') diff --git a/var/spack/repos/builtin/packages/nccl-tests/package.py b/var/spack/repos/builtin/packages/nccl-tests/package.py index 2b2f1b8fab..9e1a202da2 100644 --- a/var/spack/repos/builtin/packages/nccl-tests/package.py +++ b/var/spack/repos/builtin/packages/nccl-tests/package.py @@ -31,6 +31,11 @@ class NcclTests(MakefilePackage, CudaPackage): targets.append('MPI=1') return targets + def setup_build_environment(self, env): + cuda_arch = self.spec.variants["cuda_arch"].value + cuda_gencode = " ".join(self.cuda_flags(cuda_arch)) + env.set("NVCC_GENCODE", cuda_gencode) + def install(self, spec, prefix): mkdirp(prefix.bin) install_tree('./build', prefix.bin) -- cgit v1.2.3-60-g2f50