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

set CMAKE_CUDA_ARCHITECTURES for newer cmakes in camp (#31333)

Browse Source
This commit is contained in:
Chris White
2022-06-28 17:45:45 -07:00
committed by GitHub
parent 8bdc506579
commit 62cac911e0
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
var/spack/repos/builtin/packages/camp/package.py
View File

@@ -46,6 +46,7 @@ def cmake_args(self):
if not spec.satisfies('cuda_arch=none'): if not spec.satisfies('cuda_arch=none'):
cuda_arch = spec.variants['cuda_arch'].value cuda_arch = spec.variants['cuda_arch'].value
options.append('-DCMAKE_CUDA_ARCHITECTURES={0}'.format(cuda_arch[0]))
options.append('-DCUDA_ARCH=sm_{0}'.format(cuda_arch[0])) options.append('-DCUDA_ARCH=sm_{0}'.format(cuda_arch[0]))
flag = '-arch sm_{0}'.format(cuda_arch[0]) flag = '-arch sm_{0}'.format(cuda_arch[0])
options.append('-DCMAKE_CUDA_FLAGS:STRING={0}'.format(flag)) options.append('-DCMAKE_CUDA_FLAGS:STRING={0}'.format(flag))