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

Fix LBANN and related packages CMake CUDA arch flag (#25062)

Browse Source
This commit is contained in:
romerojosh 2021-07-26 00:16:38 -07:00 committed by GitHub
parent 63e04ce220
commit 7339f2d476
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 8 additions and 8 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

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

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

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

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

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

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

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