From 15689eac562f4d1d0edcac0ed0e75a7bca6a23ee Mon Sep 17 00:00:00 2001 From: pangyoki Date: Wed, 8 Mar 2023 19:45:07 +0800 Subject: [PATCH] delete sm60 arch for cuda11 and delete sm35 arch for cuda10 (#50316) --- cmake/cuda.cmake | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cmake/cuda.cmake b/cmake/cuda.cmake index 2f6b5d7759c..74580513213 100644 --- 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() -- GitLab