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

lbann, aluminum, hydrogen, dihydrogen: pass cuda_arch to cmake (#24074)

Browse Source
This commit is contained in:
Brian Van Essen 2021-06-02 01:02:23 -07:00 committed by GitHub
parent 622223823e
commit 3356c5fc71
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
var/spack/repos/builtin/packages/aluminum/package.py
View File

@ -66,6 +66,10 @@ def cmake_args(self):
if '+cuda' in spec:
args.append('-DCMAKE_CUDA_STANDARD=14')
archs = spec.variants['cuda_arch'].value
if archs != 'none':
arch_str = ",".join(archs)
args.append('-DCMAKE_CUDA_ARCHITECTURES=%s' % arch_str)
if spec.satisfies('@0.5:'):
args.extend([

4
var/spack/repos/builtin/packages/dihydrogen/package.py
View File

@ -156,6 +156,10 @@ def cmake_args(self):
args.append('-DCMAKE_CUDA_STANDARD=17')
else:
args.append('-DCMAKE_CUDA_STANDARD=14')
archs = spec.variants['cuda_arch'].value
if archs != 'none':
arch_str = ",".join(archs)
args.append('-DCMAKE_CUDA_ARCHITECTURES=%s' % arch_str)
if '+cuda' in spec or '+distconv' in spec:
args.append('-DcuDNN_DIR={0}'.format(

4
var/spack/repos/builtin/packages/hydrogen/package.py
View File

@ -161,6 +161,10 @@ def cmake_args(self):
if '+cuda' in spec:
args.append('-DCMAKE_CUDA_STANDARD=14')
archs = spec.variants['cuda_arch'].value
if archs != 'none':
arch_str = ",".join(archs)
args.append('-DCMAKE_CUDA_ARCHITECTURES=%s' % arch_str)
if '+rocm' in spec:
args.extend([

4
var/spack/repos/builtin/packages/lbann/package.py
View File

@ -316,6 +316,10 @@ def cmake_args(self):
args.append('-DCMAKE_CUDA_STANDARD=17')
else:
args.append('-DCMAKE_CUDA_STANDARD=14')
archs = spec.variants['cuda_arch'].value
if archs != 'none':
arch_str = ",".join(archs)
args.append('-DCMAKE_CUDA_ARCHITECTURES=%s' % arch_str)
if spec.satisfies('@:0.90') or spec.satisfies('@0.95:'):
args.append(