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

nlcglib: pass cuda_arch setting to kokkos dependency (#39725)

Browse Source 
When building with `+cuda`, the specified `cuda_arch` was not passed to kokkos,
leading to a wrong concretization.
This commit is contained in:
Loris Ercole 2023-11-23 17:49:00 +01:00 committed by GitHub
parent ee0d3a3be2
commit 624df2a1bb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
var/spack/repos/builtin/packages/nlcglib/package.py
View File

@ -49,9 +49,12 @@ class Nlcglib(CMakePackage, CudaPackage, ROCmPackage):
depends_on("rocblas") depends_on("rocblas")
depends_on("rocsolver") depends_on("rocsolver")
with when("+cuda"): for arch in CudaPackage.cuda_arch_values:
depends_on("kokkos+cuda+cuda_lambda+wrapper", when="%gcc") depends_on(
depends_on("kokkos+cuda") f"kokkos+cuda+cuda_lambda+wrapper cuda_arch={arch}",
when=f"%gcc +cuda cuda_arch={arch}",
)
depends_on(f"kokkos+cuda cuda_arch={arch}", when=f"+cuda cuda_arch={arch}")
def cmake_args(self): def cmake_args(self):
options = [ options = [