diff options
-rw-r--r-- | var/spack/repos/builtin/packages/nlcglib/package.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/var/spack/repos/builtin/packages/nlcglib/package.py b/var/spack/repos/builtin/packages/nlcglib/package.py index 5d899b17ed..4817606b8f 100644 --- a/var/spack/repos/builtin/packages/nlcglib/package.py +++ b/var/spack/repos/builtin/packages/nlcglib/package.py @@ -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 = [ |