zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

nccl-tests: respect cuda_arch variant during build (#21995)

Browse Source
This commit is contained in:
Andrew W Elble 2021-02-26 18:38:20 -05:00 committed by GitHub
parent f957f1936c
commit f2f58b70b1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
var/spack/repos/builtin/packages/nccl-tests/package.py
View File

@ -31,6 +31,11 @@ def build_targets(self):
targets.append('MPI=1') targets.append('MPI=1')
return targets 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): def install(self, spec, prefix):
mkdirp(prefix.bin) mkdirp(prefix.bin)
install_tree('./build', prefix.bin) install_tree('./build', prefix.bin)