File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -40,13 +40,8 @@ list(APPEND RAFT_CUDA_FLAGS --expt-extended-lambda --expt-relaxed-constexpr)
4040list (APPEND RAFT_CXX_FLAGS "-DCUDA_API_PER_THREAD_DEFAULT_STREAM" )
4141list (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
You can’t perform that action at this time.
0 commit comments