diff options
author | Satish Balay <balay@mcs.anl.gov> | 2022-10-27 10:02:36 -0500 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-10-27 17:02:36 +0200 |
commit | 6cb4a00280849c22191ac0d506eeac93db3052a5 (patch) | |
tree | ae923172849461fa569535f2f0d22d8f80c2e56f | |
parent | ecdfe0235597d5bafcc7f6c1c5b79fbfb991ee28 (diff) | |
download | spack-6cb4a00280849c22191ac0d506eeac93db3052a5.tar.gz spack-6cb4a00280849c22191ac0d506eeac93db3052a5.tar.bz2 spack-6cb4a00280849c22191ac0d506eeac93db3052a5.tar.xz spack-6cb4a00280849c22191ac0d506eeac93db3052a5.zip |
petsc+kokkos: pass in cuda_arch, rocm_arch to kokkos (#33530)
Also remove dependency on kokkos+wrapper
-rw-r--r-- | var/spack/repos/builtin/packages/petsc/package.py | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/var/spack/repos/builtin/packages/petsc/package.py b/var/spack/repos/builtin/packages/petsc/package.py index 71ba0cba75..830adff3b5 100644 --- a/var/spack/repos/builtin/packages/petsc/package.py +++ b/var/spack/repos/builtin/packages/petsc/package.py @@ -333,9 +333,20 @@ class Petsc(Package, CudaPackage, ROCmPackage): depends_on("hwloc", when="+hwloc") depends_on("kokkos", when="+kokkos") depends_on("kokkos-kernels", when="+kokkos") - depends_on("kokkos+cuda+wrapper+cuda_lambda", when="+kokkos +cuda") - depends_on("kokkos-kernels+cuda", when="+kokkos +cuda") - depends_on("kokkos+rocm", when="+kokkos +rocm") + for cuda_arch in CudaPackage.cuda_arch_values: + depends_on( + "kokkos+cuda+cuda_lambda cuda_arch=%s" % cuda_arch, + when="+kokkos +cuda cuda_arch=%s" % cuda_arch, + ) + depends_on( + "kokkos-kernels+cuda cuda_arch=%s" % cuda_arch, + when="+kokkos +cuda cuda_arch=%s" % cuda_arch, + ) + for rocm_arch in ROCmPackage.amdgpu_targets: + depends_on( + "kokkos+rocm amdgpu_target=%s" % rocm_arch, + when="+kokkos +rocm amdgpu_target=%s" % rocm_arch, + ) phases = ["configure", "build", "install"] |