From 529cf8a54a80dd8fcc1a156658965b590cf1d561 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Thu, 20 Jan 2022 19:49:26 +0800 Subject: [PATCH] Configure cub version automatically. (#7579) Note that when cub inside CUDA is being used, XGBoost performs checks on input size instead of using internal cub function to accept inputs larger than maximum integer. --- CMakeLists.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index d56bc7a0c..339d82697 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -138,7 +138,7 @@ if (USE_CUDA) add_subdirectory(${PROJECT_SOURCE_DIR}/gputreeshap) if ((${CMAKE_CUDA_COMPILER_VERSION} VERSION_GREATER_EQUAL 11.4) AND (NOT BUILD_WITH_CUDA_CUB)) - message(SEND_ERROR "`BUILD_WITH_CUDA_CUB` should be set to `ON` for CUDA >= 11.4") + set(BUILD_WITH_CUDA_CUB ON) endif () endif (USE_CUDA)