From 5fbc26e282928d0bff08551d00cd0dd92cd3db07 Mon Sep 17 00:00:00 2001
From: Sing_chan <51314274+betterpig@users.noreply.github.com>
Date: Tue, 5 Jul 2022 14:33:54 +0800
Subject: [PATCH] make Linux and windows both use external/cub 1.16.0 (#44004)

* make only win32 and 11.6 use external/cub

* unify cub version in linux and windows when cuda >= 11.6
---
 cmake/external/cub.cmake | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cmake/external/cub.cmake b/cmake/external/cub.cmake
index c6b435288e..de66e8d63d 100644
--- a/cmake/external/cub.cmake
+++ b/cmake/external/cub.cmake
@@ -24,7 +24,7 @@ set(CUB_PREFIX_DIR ${CUB_PATH})
 
 set(CUB_REPOSITORY ${GIT_URL}/NVlabs/cub.git)
 
-if(WIN32 AND ${CMAKE_CUDA_COMPILER_VERSION} GREATER_EQUAL 11.6)
+if(${CMAKE_CUDA_COMPILER_VERSION} GREATER_EQUAL 11.6)
   # cuda_11.6.2_511.65‘s own cub is 1.15.0, which will cause compiling error in windows.
   set(CUB_TAG 1.16.0)
   # cub 1.16.0 is not compitable with current thrust version
-- 
GitLab