未验证 提交 5fbc26e2 编写于 作者: S Sing_chan 提交者: GitHub

make Linux and windows both use external/cub 1.16.0 (#44004)

* make only win32 and 11.6 use external/cub

* unify cub version in linux and windows when cuda >= 11.6
上级 b918d063
...@@ -24,7 +24,7 @@ set(CUB_PREFIX_DIR ${CUB_PATH}) ...@@ -24,7 +24,7 @@ set(CUB_PREFIX_DIR ${CUB_PATH})
set(CUB_REPOSITORY ${GIT_URL}/NVlabs/cub.git) set(CUB_REPOSITORY ${GIT_URL}/NVlabs/cub.git)
if(WIN32 AND ${CMAKE_CUDA_COMPILER_VERSION} GREATER_EQUAL 11.6) if(${CMAKE_CUDA_COMPILER_VERSION} GREATER_EQUAL 11.6)
# cuda_11.6.2_511.65‘s own cub is 1.15.0, which will cause compiling error in windows. # cuda_11.6.2_511.65‘s own cub is 1.15.0, which will cause compiling error in windows.
set(CUB_TAG 1.16.0) set(CUB_TAG 1.16.0)
# cub 1.16.0 is not compitable with current thrust version # cub 1.16.0 is not compitable with current thrust version
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册