Build CUDA 12.8 package (#93)

mlc-ai · Jan 25, 2025 · 02eea82 · 02eea82
1 parent 827547a
commit 02eea82
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 10 deletions.
diff --git a/.github/workflows/wheel_manylinux_nightly.yaml b/.github/workflows/wheel_manylinux_nightly.yaml
@@ -20,17 +20,19 @@ jobs:
         # matrix of build configs
         config:
           - gpu: 'none'
-            image: 'mlcaidev/package-cpu:76cedcd'
+            image: 'mlcaidev/package-cpu:827547a'
           - gpu: 'cuda-12.2'
-            image: 'mlcaidev/package-cu122:76cedcd'
+            image: 'mlcaidev/package-cu122:827547a'
           - gpu: 'cuda-12.3'
-            image: 'mlcaidev/package-cu123:76cedcd'
+            image: 'mlcaidev/package-cu123:827547a'
           - gpu: 'cuda-12.4'
-            image: 'mlcaidev/package-cu124:ffe4237'
+            image: 'mlcaidev/package-cu124:827547a'
+          - gpu: 'cuda-12.8'
+            image: 'mlcaidev/package-cu128:827547a'
           - gpu: 'rocm-6.1'
-            image: 'mlcaidev/package-rocm61:76cedcd'
+            image: 'mlcaidev/package-rocm61:827547a'
           - gpu: 'rocm-6.2'
-            image: 'mlcaidev/package-rocm62:76cedcd'
+            image: 'mlcaidev/package-rocm62:827547a'
         exclude:
           - pkg_kind: ${{ github.event_name == 'schedule' && 'stable' || '' }}
 

diff --git a/scripts/build_mlc_ai_wheel_manylinux.sh b/scripts/build_mlc_ai_wheel_manylinux.sh
@@ -42,8 +42,8 @@ function audit_mlc_ai_wheel() {
 }
 
 TVM_PYTHON_DIR="/workspace/tvm/python"
-PYTHON_VERSIONS_CPU=("3.9" "3.10" "3.11" "3.12")
-PYTHON_VERSIONS_GPU=("3.9" "3.10" "3.11" "3.12")
+PYTHON_VERSIONS_CPU=("3.9" "3.10" "3.11" "3.12" "3.13")
+PYTHON_VERSIONS_GPU=("3.9" "3.10" "3.11" "3.12" "3.13")
 GPU_OPTIONS=("none" "cuda-11.7" "cuda-11.8" "cuda-12.1" "cuda-12.2" "cuda-12.3" "cuda-12.4" "cuda-12.8" "rocm-6.1" "rocm-6.2")
 GPU="none"
 

diff --git a/scripts/build_mlc_llm_wheel_manylinux.sh b/scripts/build_mlc_llm_wheel_manylinux.sh
@@ -42,8 +42,8 @@ function audit_mlc_llm_wheel() {
 }
 
 MLC_LLM_PYTHON_DIR="/workspace/mlc-llm/python"
-PYTHON_VERSIONS_CPU=("3.9" "3.10" "3.11" "3.12")
-PYTHON_VERSIONS_GPU=("3.9" "3.10" "3.11" "3.12")
+PYTHON_VERSIONS_CPU=("3.9" "3.10" "3.11" "3.12" "3.13")
+PYTHON_VERSIONS_GPU=("3.9" "3.10" "3.11" "3.12" "3.13")
 GPU_OPTIONS=("none" "cuda-11.7" "cuda-11.8" "cuda-12.1" "cuda-12.2" "cuda-12.3" "cuda-12.4" "cuda-12.8" "rocm-6.1" "rocm-6.2")
 GPU="none"