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

acts: pass cuda_arch to CMAKE_CUDA_ARCHITECTURES (#44397)

Browse Source
This commit is contained in:
Wouter Deconinck 2024-05-28 13:41:32 -05:00 committed by GitHub
parent 2bb20caa5f
commit 092dc96e6c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
var/spack/repos/builtin/packages/acts/package.py
View File

@ -478,6 +478,8 @@ def plugin_cmake_variant(plugin_name, spack_variant):
cuda_arch = spec.variants["cuda_arch"].value cuda_arch = spec.variants["cuda_arch"].value
if cuda_arch != "none": if cuda_arch != "none":
args.append(f"-DCUDA_FLAGS=-arch=sm_{cuda_arch[0]}") args.append(f"-DCUDA_FLAGS=-arch=sm_{cuda_arch[0]}")
arch_str = ";".join(self.spec.variants["cuda_arch"].value)
args.append(self.define("CMAKE_CUDA_ARCHITECTURES", arch_str))
args.append(self.define_from_variant("CMAKE_CXX_STANDARD", "cxxstd")) args.append(self.define_from_variant("CMAKE_CXX_STANDARD", "cxxstd"))