[CMake] Add CUDA include path

hcho3 · Jul 24, 2024 · 3bb9f6b · 3bb9f6b
1 parent 411c846
commit 3bb9f6b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cmake/Utils.cmake b/cmake/Utils.cmake
@@ -109,6 +109,9 @@ function(xgboost_set_cuda_flags target)
   target_include_directories(
     ${target} PRIVATE
     ${xgboost_SOURCE_DIR}/gputreeshap)
+  target_include_directories(
+    ${target} SYSTEM PRIVATE
+    ${CUDAToolkit_INCLUDE_DIRS})
 
   if(MSVC)
     target_compile_options(${target} PRIVATE