Add compute 90.

dmlc · Jul 19, 2023 · 780e297 · 780e297
1 parent 0897477
commit 780e297
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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")