Define THRUST_IGNORE_CUB_VERSION_CHECK for cuda 11

dmlc · Nov 2, 2020 · dbf981c · dbf981c
1 parent 058ce0c
commit dbf981c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
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)

diff --git a/src/data/ellpack_page.cu b/src/data/ellpack_page.cu
@@ -177,7 +177,7 @@ struct TupleScanOp {
 // Change the value type of thrust discard iterator so we can use it with cub
 template <typename T>
 class TypedDiscard : public thrust::discard_iterator<T> {
-public:
+ public:
   using value_type = T;  // NOLINT
 };