diff --git a/.github/workflows/linux_wheel.yaml b/.github/workflows/linux_wheel.yaml index 059d072..c6a3ecb 100644 --- a/.github/workflows/linux_wheel.yaml +++ b/.github/workflows/linux_wheel.yaml @@ -26,12 +26,12 @@ defaults: jobs: generate-matrix: - uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main + uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@linux-cpu-runner-lf with: package-type: wheel os: linux test-infra-repository: pytorch/test-infra - test-infra-ref: main + test-infra-ref: linux-cpu-runner-lf with-xpu: disable with-rocm: disable with-cuda: disable @@ -42,12 +42,12 @@ jobs: strategy: fail-fast: false name: Build and Upload Linux wheel - uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main + uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@linux-cpu-runner-lf with: repository: pytorch-labs/torchvision-extra-decoders ref: "" test-infra-repository: pytorch/test-infra - test-infra-ref: main + test-infra-ref: linux-cpu-runner-lf build-matrix: ${{ needs.generate-matrix.outputs.matrix }} pre-script: packaging/pre_build_script.sh post-script: packaging/post_build_script.sh diff --git a/packaging/post_build_script.sh b/packaging/post_build_script.sh index 2386988..ba139ea 100755 --- a/packaging/post_build_script.sh +++ b/packaging/post_build_script.sh @@ -2,7 +2,14 @@ set -ex -old="linux_x86_64" -new="manylinux_2_17_x86_64.manylinux2014_x86_64" -echo "Replacing ${old} with ${new} in wheel name" -mv dist/*${old}*.whl $(echo dist/*${old}*.whl | sed "s/${old}/${new}/") +ls dist +wheel_path=`find dist -type f -name "*.whl"` + +echo Found $wheel_path + +auditwheel -v repair --plat manylinux_2_38_x86_64 $wheel_path --exclude libtorch_python.so --exclude libc10.so --exclude libtorch.so --exclude libtorch_cpu.so --wheel-dir dist + +# old="linux_x86_64" +# new="manylinux_2_17_x86_64.manylinux2014_x86_64" +# echo "Replacing ${old} with ${new} in wheel name" +# mv dist/*${old}*.whl $(echo dist/*${old}*.whl | sed "s/${old}/${new}/") diff --git a/packaging/pre_build_script.sh b/packaging/pre_build_script.sh index 52df13b..3e5ea3e 100644 --- a/packaging/pre_build_script.sh +++ b/packaging/pre_build_script.sh @@ -1,3 +1,4 @@ #!/bin/bash +pip install auditwheel conda install libavif libheif -c conda-forge -yq