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

kokkos: add hip_relocatable_device_code variant. trilinos: kokkos should enable relocatable device code if requested for trilinos and it also requires the kokkos libraries be static. (#48143)

Browse Source
This commit is contained in:
Jon Rood 2024-12-17 19:48:35 -07:00 committed by GitHub
parent 3da04ccb19
commit a3bed44bf5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
var/spack/repos/builtin/packages/kokkos/package.py
View File

@ -185,6 +185,7 @@ class Kokkos(CMakePackage, CudaPackage, ROCmPackage):
"cuda_lambda": [False, "Activate experimental lambda features"],
"cuda_ldg_intrinsic": [False, "Use CUDA LDG intrinsics"],
"cuda_relocatable_device_code": [False, "Enable RDC for CUDA"],
"hip_relocatable_device_code": [False, "Enable RDC for HIP"],
"cuda_uvm": [False, "Enable unified virtual memory (UVM) for CUDA"],
"debug": [False, "Activate extra debug features - may increase compiletimes"],
"debug_bounds_check": [False, "Use bounds checking - will increase runtime"],

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

@ -410,6 +410,10 @@ class Trilinos(CMakePackage, CudaPackage, ROCmPackage):
with when("@14.4: +kokkos"):
depends_on("kokkos+wrapper", when="+wrapper")
depends_on("kokkos~wrapper", when="~wrapper")
depends_on("kokkos+cuda_relocatable_device_code~shared", when="+cuda_rdc")
depends_on("kokkos+hip_relocatable_device_code~shared", when="+rocm_rdc")
depends_on("kokkos-kernels~shared", when="+cuda_rdc")
depends_on("kokkos-kernels~shared", when="+rocm_rdc")
depends_on("kokkos~complex_align")
depends_on("kokkos@4.5.00", when="@master:")
depends_on("kokkos@4.3.01", when="@16")