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

trilinos@14.4.0 +kokkos: use external kokkos (#39397)

Browse Source
This commit is contained in:
eugeneswalker 2023-08-14 11:29:13 -07:00 committed by GitHub
parent 6e31676b29
commit a4ac3f2767
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
var/spack/repos/builtin/packages/trilinos/package.py
View File

@ -368,6 +368,21 @@ class Trilinos(CMakePackage, CudaPackage, ROCmPackage):
# ###################### Dependencies ##########################
# External Kokkos
depends_on("kokkos@4.1.00", when="@14.4.0: +kokkos")
depends_on("kokkos +wrapper", when="trilinos@14.4.0: +kokkos +wrapper")
depends_on("kokkos ~wrapper", when="trilinos@14.4.0: +kokkos ~wrapper")
for a in CudaPackage.cuda_arch_values:
arch_str = "+cuda cuda_arch=" + a
kokkos_spec = "kokkos@4.1.00 " + arch_str
depends_on(kokkos_spec, when="@14.4.0 +kokkos " + arch_str)
for a in ROCmPackage.amdgpu_targets:
arch_str = "+rocm amdgpu_target={0}".format(a)
kokkos_spec = "kokkos@4.1.00 {0}".format(arch_str)
depends_on(kokkos_spec, when="@14.4.0 +kokkos {0}".format(arch_str))
depends_on("adios2", when="+adios2")
depends_on("blas")
depends_on("boost+graph+math+exception+stacktrace", when="+boost")
@ -778,6 +793,10 @@ def define_tpl(trilinos_name, spack_name, have_dep):
for tpl_name, dep_name in tpl_dep_map:
define_tpl(tpl_name, dep_name, dep_name in spec)
# External Kokkos
if spec.satisfies("@14.4.0 +kokkos"):
options.append(define_tpl_enable("Kokkos"))
# MPI settings
options.append(define_tpl_enable("MPI"))
if "+mpi" in spec: