workflow: build cuda package

laiwenzh · Dec 12, 2024 · 26d35df · 26d35df
1 parent 7489689
commit 26d35df
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 5 deletions.
diff --git a/.github/workflows/release_packages.yml b/.github/workflows/release_packages.yml
@@ -16,12 +16,14 @@ jobs:
     strategy:
       matrix:
         arch: [X64, ARM64]
-        image: ["dev-centos7-x86:v2", "dev-centos8-arm:v2"]
+        image: ["dev-centos7-x86:v2", "dev-centos7-cu124:v1", "dev-centos8-arm:v2"]
         exclude:
           - arch: X64
             image: "dev-centos8-arm:v2"
           - arch: ARM64
             image: "dev-centos7-x86:v2"
+          - arch: ARM64
+            image: "dev-centos7-cu124:v1"
     runs-on: [self-hosted, Linux, "${{ matrix.arch }}"]
     container:
       image: registry-1.docker.io/dashinfer/${{ matrix.image }}
@@ -80,12 +82,14 @@ jobs:
     strategy:
       matrix:
         arch: [X64, ARM64]
-        image: ["dev-centos7-x86:v2", "dev-centos8-arm:v2"]
+        image: ["dev-centos7-x86:v2", "dev-centos7-cu124:v1", "dev-centos8-arm:v2"]
         exclude:
           - arch: X64
             image: "dev-centos8-arm:v2"
           - arch: ARM64
             image: "dev-centos7-x86:v2"
+          - arch: ARM64
+            image: "dev-centos7-cu124:v1"
     runs-on: [self-hosted, Linux, "${{ matrix.arch }}"]
     container:
       image: registry-1.docker.io/dashinfer/${{ matrix.image }}
@@ -110,7 +114,15 @@ jobs:
         TAG_NAME=$(git describe --tags $(git rev-list --tags --max-count=1))
         source /root/.bashrc
         VERSION_NUMBER=$(echo "$TAG_NAME" | sed 's/^v//')
-        AS_RELEASE_VERSION=$VERSION_NUMBER bash scripts/release/python_manylinux_build.sh
+        export AS_RELEASE_VERSION=$VERSION_NUMBER
+
+        if command -v nvcc &> /dev/null
+        then
+            export AS_CUDA_SM="'70;75;80;86;89;90a'"
+            bash scripts/release/python_manylinux_build_cuda.sh
+        else
+            bash scripts/release/python_manylinux_build.sh
+        fi
 
     - name: Upload wheels
       uses: actions/upload-artifact@v3

diff --git a/scripts/release/python_manylinux_build.sh b/scripts/release/python_manylinux_build.sh
@@ -2,7 +2,7 @@
 set -e -x
 
 # ALL_VERSION="3.8 3.9 3.10 3.11"
-ALL_VERSION="3.10"
+ALL_VERSION="3.8"
 BUILD_VERSION=${@:-$ALL_VERSION}
 
 echo " going to build python wheels with version: ${BUILD_VERSION}"
@@ -16,6 +16,8 @@ pushd $SCRIPT_DIR
 # 捕获arch命令的输出
 architecture=$(arch)
 
+export AS_PYTHON_PKG_NAME="dashinfer-cpu"
+
 # 使用if-else结构进行条件判断
 if [ "${architecture}" == "aarch64" ]; then
     export PLAT=manylinux_2_28_aarch64

diff --git a/scripts/release/python_manylinux_build_cuda.sh b/scripts/release/python_manylinux_build_cuda.sh
@@ -2,7 +2,7 @@
 set -e -x
 
 # ALL_VERSION="3.8 3.9 3.10 3.11"
-ALL_VERSION="3.10"
+ALL_VERSION="3.8"
 BUILD_VERSION=${@:-$ALL_VERSION}
 CUDA_VERSION=12.4