From 3f9a7b244a3a879bd935599c290b3472ea2ba7ec Mon Sep 17 00:00:00 2001 From: atalman Date: Mon, 27 Jan 2025 10:33:03 -0800 Subject: [PATCH 1/2] Test auditwheel --- .github/workflows/build-wheels-aarch64-linux.yml | 2 ++ packaging/post_build_script.sh | 6 ++++++ 2 files changed, 8 insertions(+) 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..7f3ed474bd9 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 pytorch/vision/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 From f47b6c54cdc07a2544453cc2b2079dfdcc919237 Mon Sep 17 00:00:00 2001 From: atalman Date: Mon, 27 Jan 2025 11:12:29 -0800 Subject: [PATCH 2/2] test --- packaging/post_build_script.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/packaging/post_build_script.sh b/packaging/post_build_script.sh index 7f3ed474bd9..b244cd8d1be 100644 --- a/packaging/post_build_script.sh +++ b/packaging/post_build_script.sh @@ -2,7 +2,7 @@ pip install auditwheel if [[ ${ARCH} == "aarch64" ]]; then - auditwheel repair --plat manylinux_2_28_aarch64 pytorch/vision/dist/*.whl + 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