Skip to content

Commit

Permalink
update test-infra to release version
Browse files Browse the repository at this point in the history
Summary:

pytorch/test-infra#6016 landed recently which is
breaking our ROCm builds

We point to a special branch of test-infra created just before this PR
to unblock the v0.7.0 release.

Test Plan: CI

Reviewers:

Subscribers:

Tasks:

Tags:
  • Loading branch information
vkuzo committed Dec 9, 2024
1 parent f04aec7 commit 6d50691
Show file tree
Hide file tree
Showing 4 changed files with 12 additions and 12 deletions.
4 changes: 2 additions & 2 deletions .github/workflows/build-wheels_m1.yml
Original file line number Diff line number Diff line change
Expand Up @@ -22,15 +22,15 @@ permissions:
contents: read
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@torchao-release0.7
with:
package-type: wheel
os: macos-arm64
build:
needs: generate-matrix
if: github.repository_owner == 'pytorch'
name: pytorch/ao
uses: pytorch/test-infra/.github/workflows/build_wheels_macos.yml@main
uses: pytorch/test-infra/.github/workflows/build_wheels_macos.yml@torchao-release0.7
with:
repository: pytorch/ao
ref: ${{ github.head_ref || github.ref_name }}
Expand Down
6 changes: 3 additions & 3 deletions .github/workflows/build_wheels_aarch64_linux.yml
Original file line number Diff line number Diff line change
Expand Up @@ -22,20 +22,20 @@ on:

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@torchao-release0.7
with:
package-type: wheel
os: linux-aarch64
test-infra-repository: pytorch/test-infra
test-infra-ref: main
test-infra-ref: torchao-release0.7
with-cuda: disable

build:
needs: generate-matrix
permissions:
id-token: write
contents: read
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@torchao-release0.7
with:
# Set the ref to an empty string instead of the default nightly because
# torchao doesn't have nightly branch setup yet, instead the build is
Expand Down
6 changes: 3 additions & 3 deletions .github/workflows/build_wheels_linux.yml
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ on:

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@torchao-release0.7
with:
package-type: wheel
os: linux
Expand All @@ -36,15 +36,15 @@ jobs:
permissions:
id-token: write
contents: read
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@torchao-release0.7
with:
# Set the ref to an empty string instead of the default nightly because
# torchao doesn't have nightly branch setup yet, instead the build is
# triggered daily from main with a schedule
repository: pytorch/ao
ref: ""
test-infra-repository: pytorch/test-infra
test-infra-ref: main
test-infra-ref: torchao-release0.7
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
env-var-script: packaging/env_var_script_linux.sh
pre-script: packaging/pre_build_script.sh
Expand Down
8 changes: 4 additions & 4 deletions .github/workflows/build_wheels_windows.yml
Original file line number Diff line number Diff line change
Expand Up @@ -25,12 +25,12 @@ permissions:

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@torchao-release0.7
with:
package-type: wheel
os: windows
test-infra-repository: pytorch/test-infra
test-infra-ref: main
test-infra-ref: torchao-release0.7
with-xpu: enable
with-cuda: disable

Expand All @@ -47,12 +47,12 @@ jobs:
smoke-test-script: packaging/smoke_test.py
package-name: torchao
name: ${{ matrix.repository }}
uses: pytorch/test-infra/.github/workflows/build_wheels_windows.yml@main
uses: pytorch/test-infra/.github/workflows/build_wheels_windows.yml@torchao-release0.7
with:
repository: ${{ matrix.repository }}
ref: ""
test-infra-repository: pytorch/test-infra
test-infra-ref: main
test-infra-ref: torchao-release0.7
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
pre-script: ${{ matrix.pre-script }}
env-script: ${{ matrix.env-script }}
Expand Down

0 comments on commit 6d50691

Please sign in to comment.