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

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

View file

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

View file

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

View file

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

View file

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