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})