workflow: build wheels and tgz

laiwenzh · Dec 12, 2024 · 7489689 · 7489689
1 parent 21b5883
commit 7489689
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/.github/workflows/release_packages.yml b/.github/workflows/release_packages.yml
@@ -50,8 +50,7 @@ jobs:
         git fetch --tags
         TAG_NAME=$(git describe --tags $(git rev-list --tags --max-count=1))
         VERSION_NUMBER=$(echo "$TAG_NAME" | sed 's/^v//')
-        source /opt/rh/devtoolset-7/enable
-        source activate ds_py
+        source /root/.bashrc
 
         if command -v nvcc &> /dev/null
         then
@@ -109,6 +108,7 @@ jobs:
       run: |
         git fetch --tags
         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
 
@@ -153,7 +153,7 @@ jobs:
 
   publish:
     runs-on: [self-hosted, Linux]
-    needs: [build-tgz]
+    needs: [build-tgz, build-wheels]
     strategy:
       matrix:
         arch: [X64, ARM64]

diff --git a/scripts/release/python_manylinux_build.sh b/scripts/release/python_manylinux_build.sh
@@ -60,7 +60,7 @@ build_wheel_for_python() {
     pip install -r ${REPO_ROOT}/python/requirements_dev_cpu.txt -i https://mirrors.aliyun.com/pypi/simple/
     ln -sf ${REPO_ROOT}/python/dashinfer .
     # python ${REPO_ROOT}/python/setup.py bdist_wheel
-    pip wheel ${REPO_ROOT}/python --no-deps -w ${REPO_ROOT}/python/wheelhouse/ --log wheel_log.txt
+    pip wheel ${REPO_ROOT}/python --no-deps -w ${REPO_ROOT}/python/wheelhouse/ --verbose
 
     conda deactivate
     # conda remove --name "$env_name" --all -y

diff --git a/scripts/release/python_manylinux_build_cuda.sh b/scripts/release/python_manylinux_build_cuda.sh
@@ -37,7 +37,7 @@ function repair_wheel {
         echo "Skipping non-platform wheel $wheel"
     else
         # TODO: add lib path to build lib path
-        auditwheel repair "$wheel" --plat "$PLAT" -w ${REPO_ROOT}/python/wheelhouse/ # --exclude libflash-attn.so
+        auditwheel repair "$wheel" --plat "$PLAT" -w ${REPO_ROOT}/python/wheelhouse/ --verbose
     fi
 }