Adjusted CUDA arches for NEW_RELEASE_ALL (PaddlePaddle#40660)

yuancc123456 · Mar 22, 2022 · 71b813f · 71b813f
1 parent fcf8758
commit 71b813f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 52 60 61 70 75 80")
+  set(paddle_known_gpu_archs11 "35 50 60 61 70 75 80")
 elseif(NEW_RELEASE_PYPI)
   message("Using New Release Strategy - Cubin Packge")
   add_definitions(-DNEW_RELEASE_PYPI)