hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages 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')
archs = spec.variants['cuda_arch'].value
if archs != 'none':
arch_str = ",".join(archs)
args.append('-DCMAKE_CUDA_ARCHITECTURES=%s' % arch_str)
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

@ -159,8 +159,8 @@ def cmake_args(self):
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)
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

@ -164,8 +164,8 @@ def cmake_args(self):
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)
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

@ -323,8 +323,8 @@ def cmake_args(self):
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)
arch_str = ";".join(archs)
args.append('-DCMAKE_CUDA_ARCHITECTURES=%s' % arch_str)
if spec.satisfies('@:0.90') or spec.satisfies('@0.95:'):
args.append(