From e39126ae2a96092e8add76651955f29dcc535c75 Mon Sep 17 00:00:00 2001 From: Vasiliy Kuznetsov Date: Mon, 9 Dec 2024 14:24:20 -0800 Subject: [PATCH] update test-infra to release version (#1391) * update test-infra to release version Summary: https://github.com/pytorch/test-infra/pull/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: * Update .github/workflows/build_wheels_linux.yml --------- Co-authored-by: Andrey Talman --- .github/workflows/build-wheels_m1.yml | 4 ++-- .github/workflows/build_wheels_aarch64_linux.yml | 6 +++--- .github/workflows/build_wheels_linux.yml | 7 ++++--- .github/workflows/build_wheels_windows.yml | 8 ++++---- 4 files changed, 13 insertions(+), 12 deletions(-) diff --git a/.github/workflows/build-wheels_m1.yml b/.github/workflows/build-wheels_m1.yml index 93c8086a23..9bcd0de3f2 100644 --- a/.github/workflows/build-wheels_m1.yml +++ b/.github/workflows/build-wheels_m1.yml @@ -22,7 +22,7 @@ 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 @@ -30,7 +30,7 @@ jobs: 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 }} diff --git a/.github/workflows/build_wheels_aarch64_linux.yml b/.github/workflows/build_wheels_aarch64_linux.yml index 56ea528a69..e00d07be2f 100644 --- a/.github/workflows/build_wheels_aarch64_linux.yml +++ b/.github/workflows/build_wheels_aarch64_linux.yml @@ -22,12 +22,12 @@ 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: @@ -35,7 +35,7 @@ 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 diff --git a/.github/workflows/build_wheels_linux.yml b/.github/workflows/build_wheels_linux.yml index 3c37e0e1e0..a2aa85f537 100644 --- a/.github/workflows/build_wheels_linux.yml +++ b/.github/workflows/build_wheels_linux.yml @@ -22,8 +22,9 @@ 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: + test-infra-ref: torchao-release0.7 package-type: wheel os: linux with-cpu: enable @@ -36,7 +37,7 @@ 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 @@ -44,7 +45,7 @@ jobs: 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 diff --git a/.github/workflows/build_wheels_windows.yml b/.github/workflows/build_wheels_windows.yml index bfb22cab3d..3715cb2c2e 100644 --- a/.github/workflows/build_wheels_windows.yml +++ b/.github/workflows/build_wheels_windows.yml @@ -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 @@ -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 }}