未验证 提交 71b813f0 编写于 作者: Z Zhanlue Yang 提交者: GitHub

Adjusted CUDA arches for NEW_RELEASE_ALL (#40660)

上级 fcf87582
...@@ -11,7 +11,7 @@ elseif(NEW_RELEASE_ALL) ...@@ -11,7 +11,7 @@ elseif(NEW_RELEASE_ALL)
add_definitions(-DNEW_RELEASE_ALL) add_definitions(-DNEW_RELEASE_ALL)
set(paddle_known_gpu_archs "35 50 52 60 61 70 75 80 86") set(paddle_known_gpu_archs "35 50 52 60 61 70 75 80 86")
set(paddle_known_gpu_archs10 "35 50 52 60 61 70 75") set(paddle_known_gpu_archs10 "35 50 52 60 61 70 75")
set(paddle_known_gpu_archs11 "35 50 52 60 61 70 75 80") set(paddle_known_gpu_archs11 "35 50 60 61 70 75 80")
elseif(NEW_RELEASE_PYPI) elseif(NEW_RELEASE_PYPI)
message("Using New Release Strategy - Cubin Packge") message("Using New Release Strategy - Cubin Packge")
add_definitions(-DNEW_RELEASE_PYPI) add_definitions(-DNEW_RELEASE_PYPI)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册