From cade0018ea985768baa307d738e7cc249a634dfc Mon Sep 17 00:00:00 2001 From: Zhanlue Yang Date: Tue, 12 Apr 2022 11:00:26 +0800 Subject: [PATCH] Adjusted CUDA Arches (#41628) --- cmake/cuda.cmake | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cmake/cuda.cmake b/cmake/cuda.cmake index e09429bc429..75f4f192444 100644 --- 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) -- GitLab