From f421c26d3563da351f54a7baf89574521929b916 Mon Sep 17 00:00:00 2001 From: Rory Mitchell Date: Thu, 11 Aug 2022 22:36:47 +0200 Subject: [PATCH] Tune cuda architectures (#8152) --- cmake/Utils.cmake | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/cmake/Utils.cmake b/cmake/Utils.cmake index a38ed955f..9c8dc188c 100644 --- a/cmake/Utils.cmake +++ b/cmake/Utils.cmake @@ -91,21 +91,21 @@ function(format_gencode_flags flags out) # Set up architecture flags if(NOT flags) if (CUDA_VERSION VERSION_GREATER_EQUAL "11.1") - set(flags "52;60;61;70;75;80;86") + set(flags "50;60;70;80") elseif (CUDA_VERSION VERSION_GREATER_EQUAL "11.0") - set(flags "52;60;61;70;75;80") + set(flags "50;60;70;80") elseif(CUDA_VERSION VERSION_GREATER_EQUAL "10.0") - set(flags "35;50;52;60;61;70;75") + set(flags "35;50;60;70") elseif(CUDA_VERSION VERSION_GREATER_EQUAL "9.0") - set(flags "35;50;52;60;61;70") + set(flags "35;50;60;70") else() - set(flags "35;50;52;60;61") + set(flags "35;50;60") endif() endif() if (CMAKE_VERSION VERSION_GREATER_EQUAL "3.18") cmake_policy(SET CMP0104 NEW) - list(POP_BACK flags latest_arch) + list(GET flags -1 latest_arch) list(TRANSFORM flags APPEND "-real") list(APPEND flags ${latest_arch}) set(CMAKE_CUDA_ARCHITECTURES ${flags})