[CMake] Explicitly link with CCCL (standalone or CTK) (#10624)

2024-07-23 18:42:54 -07:00
parent 7949a8d5f4
commit 411c8466bd
3 changed files with 21 additions and 3 deletions
--- a/cmake/Utils.cmake
+++ b/cmake/Utils.cmake
@@ -104,11 +104,11 @@ function(xgboost_set_cuda_flags target)
    target_compile_definitions(${target} PRIVATE -DXGBOOST_USE_NVTX=1)
  endif()

+  target_link_libraries(${target} PRIVATE CCCL::CCCL)
  target_compile_definitions(${target} PRIVATE -DXGBOOST_USE_CUDA=1)
  target_include_directories(
    ${target} PRIVATE
-    ${xgboost_SOURCE_DIR}/gputreeshap
-    ${CUDAToolkit_INCLUDE_DIRS})
+    ${xgboost_SOURCE_DIR}/gputreeshap)

  if(MSVC)
    target_compile_options(${target} PRIVATE