(tvm) 09/19: enable sm90 codegen for thrust and vllm

[masahi]

This is an automated email from the ASF dual-hosted git repository.

masahi pushed a commit to branch unity
in repository https://gitbox.apache.org/repos/asf/tvm.git

commit d0634f26ac956b93c6d6e8c90e7f8d95639942c2
Author: Masahiro Masuda <masahi...@gmail.com>
AuthorDate: Sat Dec 2 01:30:20 2023 +0000

    enable sm90 codegen for thrust and vllm
---
 cmake/modules/CUDA.cmake         | 2 +-
 cmake/modules/contrib/vllm.cmake | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/cmake/modules/CUDA.cmake b/cmake/modules/CUDA.cmake
index f2370648e1..1a8d73668f 100644
--- a/cmake/modules/CUDA.cmake
+++ b/cmake/modules/CUDA.cmake
@@ -67,7 +67,7 @@ if(USE_CUDA)
 
     if(NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
       message(WARNING "CMAKE_CUDA_ARCHITECTURES not set, compiling Thrust for 
sm80 and sm75.")
-      set(CMAKE_CUDA_ARCHITECTURES "80;75")
+      set(CMAKE_CUDA_ARCHITECTURES "90;80")
     endif()
 
     set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} --expt-extended-lambda")
diff --git a/cmake/modules/contrib/vllm.cmake b/cmake/modules/contrib/vllm.cmake
index ae9474cb8b..64238e2360 100644
--- a/cmake/modules/contrib/vllm.cmake
+++ b/cmake/modules/contrib/vllm.cmake
@@ -22,7 +22,7 @@ if(USE_VLLM)
 
   if(NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
     message(WARNING "CMAKE_CUDA_ARCHITECTURES not set, compiling vLLM kernels 
for sm80 and sm75.")
-    set(CMAKE_CUDA_ARCHITECTURES "80;75")
+    set(CMAKE_CUDA_ARCHITECTURES "90;80")
   endif()
 
   tvm_file_glob(GLOB VLLM_CONTRIB_SRC src/runtime/contrib/vllm/*.cu 
src/runtime/contrib/vllm/*.cc)