From 3bb9f6b1d7c1e680c5a976d5389dd0c5d545e7b4 Mon Sep 17 00:00:00 2001
From: Hyunsu Cho <chohyu01@cs.washington.edu>
Date: Tue, 23 Jul 2024 19:46:34 -0700
Subject: [PATCH] [CMake] Add CUDA include path

---
 cmake/Utils.cmake | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/cmake/Utils.cmake b/cmake/Utils.cmake
index a00bb7a9239e..bed0c43e78a9 100644
--- 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