From 8c675c0551605ca38ea0453d02ab3ffc3c069f36 Mon Sep 17 00:00:00 2001 From: nhzlx Date: Thu, 31 Oct 2019 02:17:46 +0000 Subject: [PATCH] add compute Capability for nano, agx xavier test=develop --- cmake/cuda.cmake | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/cmake/cuda.cmake b/cmake/cuda.cmake index d571925b20..9ff908a4c8 100644 --- a/cmake/cuda.cmake +++ b/cmake/cuda.cmake @@ -4,9 +4,9 @@ endif() set(paddle_known_gpu_archs "30 35 50 52 60 61 70") set(paddle_known_gpu_archs7 "30 35 50 52") -set(paddle_known_gpu_archs8 "30 35 50 52 60 61") -set(paddle_known_gpu_archs9 "30 35 50 52 60 61 62 70") -set(paddle_known_gpu_archs10 "30 35 50 52 60 61 62 70 75") +set(paddle_known_gpu_archs8 "30 35 50 52 53 60 61 62") +set(paddle_known_gpu_archs9 "30 35 50 52 53 60 61 62 70") +set(paddle_known_gpu_archs10 "30 35 50 52 53 60 61 62 70 72 75") ###################################################################################### # A function for automatic detection of GPUs installed (if autodetection is enabled) -- GitLab