Include CUDA 90a in package build for CUTLASS kernels

mlc-ai · Sep 28, 2024 · 3525748 · 3525748
1 parent cd9e596
commit 3525748
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/scripts/build_mlc_ai_wheel_manylinux.sh b/scripts/build_mlc_ai_wheel_manylinux.sh
@@ -101,7 +101,7 @@ echo set\(USE_VULKAN ON\) >>config.cmake
 if [[ ${GPU} == cuda-11.7 ]]; then
 	CUDA_ARCHS="80;86;87"
 elif [[ ${GPU} == cuda* ]]; then
-	CUDA_ARCHS="80;86;87;89;90"
+	CUDA_ARCHS="80;86;87;89;90;90a"
 fi
 
 if [[ ${GPU} == rocm* ]]; then

diff --git a/scripts/build_mlc_llm_wheel_manylinux.sh b/scripts/build_mlc_llm_wheel_manylinux.sh
@@ -98,7 +98,7 @@ echo set\(USE_VULKAN ON\) >>config.cmake
 if [[ ${GPU} == cuda-11.7 ]]; then
 	CUDA_ARCHS="80;86;87"
 elif [[ ${GPU} == cuda* ]]; then
-	CUDA_ARCHS="80;86;87;89;90"
+	CUDA_ARCHS="80;86;87;89;90;90a"
 fi
 
 if [[ ${GPU} == rocm* ]]; then