Skip to content

Commit b0448e1

Browse files
authored
Release only change temporary fix aarch64 builds (#4544)
<!-- copilot:summary --> ### <samp>🤖 Generated by Copilot at 60a305c</samp> Use `--pre` option to install PyTorch 2.1.0 on aarch64 for wheel building. This enables testing and releasing vision and audio packages that depend on the latest PyTorch version. Here is the test: https://github.com/pytorch/builder/actions/runs/6041507840/job/16394612224#step:11:3131
1 parent 5294b9c commit b0448e1

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

.github/workflows/build_wheels_linux.yml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -149,6 +149,10 @@ jobs:
149149
set -euxo pipefail
150150
source "${BUILD_ENV_FILE}"
151151
# shellcheck disable=SC2086
152+
# temporary change to force rlease aarch64 vision and audio builds use pre and hence install 2.1.0 torch
153+
if [[ "${ARCH}" == "aarch64" ]]; then
154+
PIP_INSTALL_TORCH=${PIP_INSTALL_TORCH/"pip install"/"pip install --pre"}
155+
fi
152156
${CONDA_RUN} ${PIP_INSTALL_TORCH}
153157
- name: Run Pre-Script with Caching
154158
if: ${{ inputs.pre-script != '' }}

0 commit comments

Comments
 (0)