delete sm60 arch for cuda11 and delete sm35 arch for cuda10 (#50316)

15689eac · pangyoki · GitHub · 3be6791f · 15689eac
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

cmake/cuda.cmake cmake/cuda.cmake +2 -2

未找到文件。
--- a/cmake/cuda.cmake
+++ b/cmake/cuda.cmake
@@ -18,7 +18,7 @@ elseif(NEW_RELEASE_PYPI)
  add_definitions(-DNEW_RELEASE_PYPI)
  set(paddle_known_gpu_archs "35 50 52 60 61 70 75 80 86")
  set(paddle_known_gpu_archs10 "")
-  set(paddle_known_gpu_archs11 "60 61 70 75 80")
+  set(paddle_known_gpu_archs11 "61 70 75 80")
 elseif(NEW_RELEASE_JIT)
  message("Using New Release Strategy - JIT Packge")
  add_definitions(-DNEW_RELEASE_JIT)
@@ -27,7 +27,7 @@ elseif(NEW_RELEASE_JIT)
  set(paddle_known_gpu_archs11 "35 50 60 70 75 80")
 else()
  set(paddle_known_gpu_archs "35 50 52 60 61 70 75 80")
-  set(paddle_known_gpu_archs10 "35 50 52 60 61 70 75")
+  set(paddle_known_gpu_archs10 "50 52 60 61 70 75")
  set(paddle_known_gpu_archs11 "52 60 61 70 75 80")
 endif()