chai - make CUDA_SEPARABLE_COMPILATION flag optional (#42165)

This commit is contained in:
Brian Han 2024-01-18 15:53:28 -08:00 committed by GitHub
parent 8ac27241b6
commit 2d37d5474d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -62,6 +62,11 @@ class Chai(CachedCMakePackage, CudaPackage, ROCmPackage):
version("1.0", tag="v1.0", commit="501a098ad879dc8deb4a74fcfe8c08c283a10627", submodules=True)
variant("enable_pick", default=False, description="Enable pick method")
variant(
"separable_compilation",
default=True,
description="Build with CUDA_SEPARABLE_COMPILATION flag on ",
)
variant("shared", default=True, description="Build Shared Libs")
variant("raja", default=False, description="Build plugin for RAJA")
variant("benchmarks", default=False, description="Build benchmarks.")
@ -155,6 +160,7 @@ def initconfig_hardware_entries(self):
if "+cuda" in spec:
entries.append(cmake_cache_option("ENABLE_CUDA", True))
if "+separable_compilation" in spec:
entries.append(cmake_cache_option("CMAKE_CUDA_SEPARABLE_COMPILATION", True))
entries.append(cmake_cache_option("CUDA_SEPARABLE_COMPILATION", True))