From 3cc6ae69ed93388b2648bcc819d593130dede752 Mon Sep 17 00:00:00 2001
From: Sing_chan <51314274+betterpig@users.noreply.github.com>
Date: Fri, 1 Jul 2022 19:42:00 +0800
Subject: [PATCH] make only win32 and 11.6 use external/cub (#44005)

---
 cmake/third_party.cmake | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/cmake/third_party.cmake b/cmake/third_party.cmake
index 3c39d50878..fa859170e2 100644
--- a/cmake/third_party.cmake
+++ b/cmake/third_party.cmake
@@ -257,8 +257,8 @@ if(WITH_ONNXRUNTIME)
 endif()
 
 if(WITH_GPU)
-    if (${CMAKE_CUDA_COMPILER_VERSION} LESS 11.0 OR ${CMAKE_CUDA_COMPILER_VERSION}
-                                                    GREATER_EQUAL 11.6)
+    if(${CMAKE_CUDA_COMPILER_VERSION} LESS 11.0
+        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