diff options
| -rw-r--r-- | var/spack/repos/builtin/packages/nccl-tests/package.py | 5 |
1 files changed, 5 insertions, 0 deletions
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) |