未验证 提交 3cc6ae69 编写于 作者: S Sing_chan 提交者: GitHub

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

上级 fde34eb8
...@@ -257,8 +257,8 @@ if(WITH_ONNXRUNTIME) ...@@ -257,8 +257,8 @@ if(WITH_ONNXRUNTIME)
endif() endif()
if(WITH_GPU) if(WITH_GPU)
if (${CMAKE_CUDA_COMPILER_VERSION} LESS 11.0 OR ${CMAKE_CUDA_COMPILER_VERSION} if(${CMAKE_CUDA_COMPILER_VERSION} LESS 11.0
GREATER_EQUAL 11.6) OR (WIN32 AND ${CMAKE_CUDA_COMPILER_VERSION} GREATER_EQUAL 11.6))
include(external/cub) # download cub include(external/cub) # download cub
list(APPEND third_party_deps extern_cub) list(APPEND third_party_deps extern_cub)
endif() endif()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册