make only win32 and 11.6 use external/cub (#44005)

PaddlePaddle · Jul 1, 2022 · 3cc6ae6 · 3cc6ae6
1 parent fde34eb
commit 3cc6ae6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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()