diff options
| author | romerojosh <joshr@nvidia.com> | 2021-07-26 00:16:38 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-07-26 09:16:38 +0200 |
| commit | 7339f2d476cb8a4007aad12b740a78d67d086142 (patch) | |
| tree | feae30a6f6bdc64271e20a78a5e4db5fd0e3a690 | |
| parent | 63e04ce220f9ae5a135a1d00bc5fe00872aa1731 (diff) | |
| download | spack-7339f2d476cb8a4007aad12b740a78d67d086142.tar.gz spack-7339f2d476cb8a4007aad12b740a78d67d086142.tar.bz2 spack-7339f2d476cb8a4007aad12b740a78d67d086142.tar.xz spack-7339f2d476cb8a4007aad12b740a78d67d086142.zip | |
Fix LBANN and related packages CMake CUDA arch flag (#25062)
4 files changed, 8 insertions, 8 deletions
diff --git a/var/spack/repos/builtin/packages/aluminum/package.py b/var/spack/repos/builtin/packages/aluminum/package.py index 113564de5b..ea8d9e8317 100644 --- a/var/spack/repos/builtin/packages/aluminum/package.py +++ b/var/spack/repos/builtin/packages/aluminum/package.py @@ -69,8 +69,8 @@ class Aluminum(CMakePackage, CudaPackage, ROCmPackage): 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([ diff --git a/var/spack/repos/builtin/packages/dihydrogen/package.py b/var/spack/repos/builtin/packages/dihydrogen/package.py index 0bd36298d2..4a133c0a05 100644 --- a/var/spack/repos/builtin/packages/dihydrogen/package.py +++ b/var/spack/repos/builtin/packages/dihydrogen/package.py @@ -159,8 +159,8 @@ class Dihydrogen(CMakePackage, CudaPackage, ROCmPackage): 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( diff --git a/var/spack/repos/builtin/packages/hydrogen/package.py b/var/spack/repos/builtin/packages/hydrogen/package.py index 7fb8f1b5f4..61394c4818 100644 --- a/var/spack/repos/builtin/packages/hydrogen/package.py +++ b/var/spack/repos/builtin/packages/hydrogen/package.py @@ -164,8 +164,8 @@ class Hydrogen(CMakePackage, CudaPackage, ROCmPackage): 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([ diff --git a/var/spack/repos/builtin/packages/lbann/package.py b/var/spack/repos/builtin/packages/lbann/package.py index 2471850790..e1a965c616 100644 --- a/var/spack/repos/builtin/packages/lbann/package.py +++ b/var/spack/repos/builtin/packages/lbann/package.py @@ -323,8 +323,8 @@ class Lbann(CMakePackage, CudaPackage, ROCmPackage): 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( |