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
1 changed files with 2 additions and 0 deletions
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
if cuda_arch != "none":
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"))