Add CMAKE_CUDA_ARCHITECTURES to CMakeLists.txt

mlc-ai · Jan 1, 2024 · 43985e7 · 43985e7
1 parent 5c88436
commit 43985e7
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/3rdparty/cutlass_fpA_intB_gemm b/3rdparty/cutlass_fpA_intB_gemm
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -531,6 +531,15 @@ else()
   set(CMAKE_CUDA_STANDARD 17)
 endif()
 
+if ((USE_CUDA AND USE_CUTLASS))
+  if(NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
+    set(CMAKE_CUDA_ARCHITECTURES "75;80;86;89;90")
+    message(STATUS "Set CMAKE_CUDA_ARCHITECTURES=${CMAKE_CUDA_ARCHITECTURES}")
+  else ()
+    message(STATUS "Found CMAKE_CUDA_ARCHITECTURES=${CMAKE_CUDA_ARCHITECTURES}")
+  endif()
+endif()
+
 # Module rules
 include(cmake/modules/VTA.cmake)
 include(cmake/modules/StandaloneCrt.cmake)