diff --git a/cmake/Utils.cmake b/cmake/Utils.cmake
index 4147643a78fd..3d0973c745e2 100644
--- 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
index b69c19343698..d560acc6a7cc 100644
--- 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
 };