test if linux cuda11.6 and above need cub (#49404)

f65ca8ca · zqw_1997 · GitHub · e3f39833 · f65ca8ca
隐藏空白更改
内联并排

Showing with 1 addition and 2 deletion

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

未找到文件。
--- a/cmake/third_party.cmake
+++ b/cmake/third_party.cmake
@@ -321,8 +321,7 @@ endif()

 if(WITH_GPU)
  if(${CMAKE_CUDA_COMPILER_VERSION} LESS 11.0
-     OR (${CMAKE_CUDA_COMPILER_VERSION} GREATER_EQUAL 11.6
-         AND ${CMAKE_CUDA_COMPILER_VERSION} LESS 11.8))
+     OR (WIN32 AND ${CMAKE_CUDA_COMPILER_VERSION} GREATER_EQUAL 11.6))
    include(external/cub) # download cub
    list(APPEND third_party_deps extern_cub)
  endif()