Adjusted CUDA Arches (#41628)

cade0018 · Zhanlue Yang · GitHub · 0b4c3c20 · cade0018
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

cmake/cuda.cmake cmake/cuda.cmake +1 -1

未找到文件。
--- a/cmake/cuda.cmake
+++ b/cmake/cuda.cmake
@@ -11,7 +11,7 @@ elseif(NEW_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_archs10 "35 50 52 60 61 70 75")
-  set(paddle_known_gpu_archs11 "35 50 60 61 70 75 80")
+  set(paddle_known_gpu_archs11 "50 60 61 70 75 80")
 elseif(NEW_RELEASE_PYPI)
  message("Using New Release Strategy - Cubin Packge")
  add_definitions(-DNEW_RELEASE_PYPI)