From 7a0ccfbb491607a7b7dc2f13256db999b11d09cb Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Wed, 19 Jul 2023 13:42:38 +0800 Subject: [PATCH] Add compute 90. (#9397) --- cmake/Utils.cmake | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cmake/Utils.cmake b/cmake/Utils.cmake index dc523d03a..cb239f79c 100644 --- a/cmake/Utils.cmake +++ b/cmake/Utils.cmake @@ -90,8 +90,8 @@ function(format_gencode_flags flags out) endif() # Set up architecture flags if(NOT flags) - if (CUDA_VERSION VERSION_GREATER_EQUAL "11.1") - set(flags "50;60;70;80") + if (CUDA_VERSION VERSION_GREATER_EQUAL "11.8") + set(flags "50;60;70;80;90") elseif (CUDA_VERSION VERSION_GREATER_EQUAL "11.0") set(flags "50;60;70;80") elseif(CUDA_VERSION VERSION_GREATER_EQUAL "10.0")