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

3cc6ae69 · Sing_chan · GitHub · fde34eb8 · 3cc6ae69
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

cmake/third_party.cmake cmake/third_party.cmake +2 -2

未找到文件。
--- 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()