From f65ca8cada4bc8d1d65165221cb06809016e4db9 Mon Sep 17 00:00:00 2001
From: zqw_1997 <118182234+zhengqiwen1997@users.noreply.github.com>
Date: Thu, 19 Jan 2023 12:05:30 +0800
Subject: [PATCH] test if linux cuda11.6 and above need cub (#49404)

---
 cmake/third_party.cmake | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/cmake/third_party.cmake b/cmake/third_party.cmake
index 602f966cf8e..f181c60e6e9 100755
--- a/cmake/third_party.cmake
+++ b/cmake/third_party.cmake
@@ -321,8 +321,7 @@ endif()
 
 if(WITH_GPU)
   if(${CMAKE_CUDA_COMPILER_VERSION} LESS 11.0
-     OR (${CMAKE_CUDA_COMPILER_VERSION} GREATER_EQUAL 11.6
-         AND ${CMAKE_CUDA_COMPILER_VERSION} LESS 11.8))
+     OR (WIN32 AND ${CMAKE_CUDA_COMPILER_VERSION} GREATER_EQUAL 11.6))
     include(external/cub) # download cub
     list(APPEND third_party_deps extern_cub)
   endif()
-- 
GitLab