diff --git a/.github/workflows/build-wheels-aarch64-linux.yml b/.github/workflows/build-wheels-aarch64-linux.yml index 05c83991d5b..a8b3301e53e 100644 --- a/.github/workflows/build-wheels-aarch64-linux.yml +++ b/.github/workflows/build-wheels-aarch64-linux.yml @@ -2,6 +2,8 @@ name: Build Aarch64 Linux Wheels on: pull_request: + paths: + - packaging/post_build_script.sh push: branches: - nightly diff --git a/packaging/post_build_script.sh b/packaging/post_build_script.sh index 253980b98c3..b244cd8d1be 100644 --- a/packaging/post_build_script.sh +++ b/packaging/post_build_script.sh @@ -1,4 +1,10 @@ #!/bin/bash +pip install auditwheel + +if [[ ${ARCH} == "aarch64" ]]; then + auditwheel repair --plat manylinux_2_28_aarch64 dist/*.whl +fi + LD_LIBRARY_PATH="/usr/local/lib:$CUDA_HOME/lib64:$LD_LIBRARY_PATH" python packaging/wheel/relocate.py pip install torchvision-extra-decoders