Fix LBANN and related packages CMake CUDA arch flag (#25062)
This commit is contained in:
parent
63e04ce220
commit
7339f2d476
4 changed files with 8 additions and 8 deletions
|
@ -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:'):
|
||||
|
|
|
@ -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:
|
||||
|
|
|
@ -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:
|
||||
|
|
|
@ -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:'):
|
||||
|
|
Loading…
Reference in a new issue