Define THRUST_IGNORE_CUB_VERSION_CHECK for cuda 11

dmlc · Nov 2, 2020 · b29d6b3 · b29d6b3
1 parent 058ce0c
commit b29d6b3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cmake/Utils.cmake b/cmake/Utils.cmake
@@ -155,7 +155,7 @@ function(xgboost_set_cuda_flags target)
     enable_nvtx(${target})
   endif (USE_NVTX)
 
-  target_compile_definitions(${target} PRIVATE -DXGBOOST_USE_CUDA=1)
+  target_compile_definitions(${target} PRIVATE -DXGBOOST_USE_CUDA=1 -DTHRUST_IGNORE_CUB_VERSION_CHECK=1)
   target_include_directories(${target} PRIVATE ${xgboost_SOURCE_DIR}/cub/)
 
   if (MSVC)