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

trilinos: add cuda_constexpr variant (#45812)

Browse Source
This commit is contained in:
Daniel Arndt 2024-09-24 08:26:16 -04:00 committed by GitHub
parent c070ddac97
commit 728da2ff87
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -80,6 +80,11 @@ class Trilinos(CMakePackage, CudaPackage, ROCmPackage):
# Build options # Build options
variant("complex", default=False, description="Enable complex numbers in Trilinos") variant("complex", default=False, description="Enable complex numbers in Trilinos")
variant(
"cuda_constexpr",
default=False,
description="Enable relaxed constexpr functions for CUDA build",
)
variant("cuda_rdc", default=False, description="Turn on RDC for CUDA build") variant("cuda_rdc", default=False, description="Turn on RDC for CUDA build")
variant("rocm_rdc", default=False, description="Turn on RDC for ROCm build") variant("rocm_rdc", default=False, description="Turn on RDC for ROCm build")
variant( variant(
@ -1006,6 +1011,7 @@ def define_tpl(trilinos_name, spack_name, have_dep):
[ [
define_kok_enable("CUDA_UVM", use_uvm), define_kok_enable("CUDA_UVM", use_uvm),
define_kok_enable("CUDA_LAMBDA", True), define_kok_enable("CUDA_LAMBDA", True),
define_kok_enable("CUDA_CONSTEXPR", "cuda_constexpr"),
define_kok_enable("CUDA_RELOCATABLE_DEVICE_CODE", "cuda_rdc"), define_kok_enable("CUDA_RELOCATABLE_DEVICE_CODE", "cuda_rdc"),
] ]
) )