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

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

View File

@ -49,9 +49,12 @@ class Nlcglib(CMakePackage, CudaPackage, ROCmPackage):
depends_on("rocblas")
depends_on("rocsolver")
with when("+cuda"):
depends_on("kokkos+cuda+cuda_lambda+wrapper", when="%gcc")
depends_on("kokkos+cuda")
for arch in CudaPackage.cuda_arch_values:
depends_on(
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):
options = [