Skip to content

Commit ba3b0a4

Browse files
committed
Override the torch pip package version when building wheels
This override will ensure that the release/0.2 branch always builds against torch==2.3.0 instead of trying to use newer versions or nightlies.
1 parent d4ccfa6 commit ba3b0a4

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

.github/workflows/build-wheels-linux.yml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,9 @@ jobs:
5050
test-infra-repository: pytorch/test-infra
5151
test-infra-ref: release/2.3
5252
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
53+
# All wheels built in the release/0.2 branch should use this version of
54+
# the torch pip package.
55+
pytorch-version-override: 2.3.0
5356
# ExecuTorch only needs the first layer of submodules; override the
5457
# "recursive" default to do less work, and to give the buck daemon fewer
5558
# files to look at.

.github/workflows/build-wheels-m1.yml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,9 @@ jobs:
5050
test-infra-repository: pytorch/test-infra
5151
test-infra-ref: release/2.3
5252
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
53+
# All wheels built in the release/0.2 branch should use this version of
54+
# the torch pip package.
55+
pytorch-version-override: 2.3.0
5356
# ExecuTorch only needs the first layer of submodules; override the
5457
# "recursive" default to do less work, and to give the buck daemon fewer
5558
# files to look at.

0 commit comments

Comments
 (0)