Set test-infra-ref param for release (#1739)

* Set test-infra-ref param for release

* rest

Author: atalman

.github/workflows/build-wheels_m1.yml

@@ -26,6 +26,7 @@ jobs:
     with:
       package-type: wheel
       os: macos-arm64
+      test-infra-ref: torchao-release0.9
   build:
     needs: generate-matrix
     if: github.repository_owner == 'pytorch'
@@ -36,6 +37,7 @@ jobs:
       ref: ${{ github.head_ref || github.ref_name }}
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
       pre-script: packaging/pre_build_script.sh
+      test-infra-ref: torchao-release0.9
       # post-script: packaging/post_build_script.sh
       package-name: torchao
       runner-type: macos-m1-stable
@@ -70,5 +72,5 @@ jobs:
             Error Information:
             ${{ needs.generate-matrix.result == 'failure' && 'Matrix generation failed' || '' }}
             ${{ needs.build.result == 'failure' && 'Build job failed' || '' }}
-            
+
             This is an automated notification. Please check the GitHub Actions page for more details about the failure.

.github/workflows/build_wheels_aarch64_linux.yml

@@ -27,7 +27,7 @@ jobs:
       package-type: wheel
       os: linux-aarch64
       test-infra-repository: pytorch/test-infra
-      test-infra-ref: main
+      test-infra-ref: torchao-release0.9
       with-cuda: disable
       # please note: excluding 3.13t for aarch64 builds for now
       python-versions: '["3.9", "3.10", "3.11", "3.12", "3.13"]'
@@ -48,6 +48,7 @@ jobs:
       pre-script: packaging/pre_build_script.sh
       # post-script: packaging/post_build_script.sh
       smoke-test-script: packaging/smoke_test.py
+      test-infra-ref: torchao-release0.9
       package-name: torchao
       trigger-event: ${{ github.event_name }}
       architecture: aarch64
@@ -83,5 +84,5 @@ jobs:
             Error Information:
             ${{ needs.generate-matrix.result == 'failure' && 'Matrix generation failed' || '' }}
             ${{ needs.build.result == 'failure' && 'Build job failed' || '' }}
-            
+
             This is an automated notification. Please check the GitHub Actions page for more details about the failure.

.github/workflows/build_wheels_linux.yml

@@ -30,6 +30,7 @@ jobs:
       with-cuda: enable
       with-rocm: enable
       with-xpu: enable
+      test-infra-ref: torchao-release0.9
       # please note: excluding 3.13t for aarch64 builds for now
       python-versions: '["3.9", "3.10", "3.11", "3.12", "3.13"]'
 
@@ -46,7 +47,7 @@ jobs:
       repository: pytorch/ao
       ref: ""
       test-infra-repository: pytorch/test-infra
-      test-infra-ref: main
+      test-infra-ref: torchao-release0.9
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
       env-var-script: packaging/env_var_script_linux.sh
       pre-script: packaging/pre_build_script.sh

.github/workflows/build_wheels_windows.yml

@@ -30,7 +30,7 @@ jobs:
       package-type: wheel
       os: windows
       test-infra-repository: pytorch/test-infra
-      test-infra-ref: main
+      test-infra-ref: torchao-release0.9
       with-xpu: enable
       with-cuda: disable
 
@@ -52,7 +52,7 @@ jobs:
       repository: ${{ matrix.repository }}
       ref: ""
       test-infra-repository: pytorch/test-infra
-      test-infra-ref: main
+      test-infra-ref: torchao-release0.9
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
       pre-script: ${{ matrix.pre-script }}
       env-script: ${{ matrix.env-script }}