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

petsc+kokkos: pass in cuda_arch, rocm_arch to kokkos (#33530)

Browse Source 
Also remove dependency on kokkos+wrapper
This commit is contained in:
Satish Balay 2022-10-27 10:02:36 -05:00 committed by GitHub
parent ecdfe02355
commit 6cb4a00280
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
var/spack/repos/builtin/packages/petsc/package.py
View File

@ -333,9 +333,20 @@ def check_fortran_compiler(self):
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"]