Skip to content

Commit 7967481

Browse files
Use rapids_cuda_enable_fatbin_compression (#2780)
Standardize compression flags via rapids_cuda_enable_fatbin_compression Authors: - Robert Maynard (https://github.com/robertmaynard) Approvers: - Bradley Dice (https://github.com/bdice) URL: #2780
1 parent 44241b3 commit 7967481

1 file changed

Lines changed: 2 additions & 7 deletions

File tree

cpp/cmake/modules/ConfigureCUDA.cmake

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -40,13 +40,8 @@ list(APPEND RAFT_CUDA_FLAGS --expt-extended-lambda --expt-relaxed-constexpr)
4040
list(APPEND RAFT_CXX_FLAGS "-DCUDA_API_PER_THREAD_DEFAULT_STREAM")
4141
list(APPEND RAFT_CUDA_FLAGS "-DCUDA_API_PER_THREAD_DEFAULT_STREAM")
4242
# make sure we produce smallest binary size
43-
list(APPEND RAFT_CUDA_FLAGS -Xfatbin=-compress-all)
44-
if(CMAKE_CUDA_COMPILER_ID STREQUAL "NVIDIA"
45-
AND (CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL 12.9 AND CMAKE_CUDA_COMPILER_VERSION
46-
VERSION_LESS 13.0)
47-
)
48-
list(APPEND RAFT_CUDA_FLAGS -Xfatbin=--compress-level=3)
49-
endif()
43+
include(${rapids-cmake-dir}/cuda/enable_fatbin_compression.cmake)
44+
rapids_cuda_enable_fatbin_compression(VARIABLE RAFT_CUDA_FLAGS TUNE_FOR rapids)
5045

5146
# Option to enable line info in CUDA device compilation to allow introspection when profiling /
5247
# memchecking

0 commit comments

Comments
 (0)