Make sure we are building against test channel for release (#6168)

* Make sure we are building against test chanell for release

* Cleanup

* Cleanup

* Refactoring logic

* Remove tagged tests

Author: atalman

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

@@ -7,6 +7,7 @@ on:
     branches:
       - nightly
       - main
+      - release/*
     tags:
         # NOTE: Binary build pipelines should only get triggered on release candidate builds
         # Release candidate tags look like: v1.11.0-rc1
@@ -31,6 +32,10 @@ jobs:
           if [[ ${GITHUB_REF_NAME} = *-rc[0-9]* ]]; then
             echo "CHANNEL=test" >> "$GITHUB_ENV"
           fi
+      - name: Set Release CHANNEL (for release)
+        if: ${{ (github.event_name == 'pull_request' && startsWith(github.base_ref, 'release')) || startsWith(github.ref, 'release') }}
+        run: |
+          echo "CHANNEL=test" >> "$GITHUB_ENV"
       - name: Build TorchVision M1 wheel
         shell: arch -arch arm64 bash {0}
         env:
@@ -104,6 +109,10 @@ jobs:
           if [[ ${GITHUB_REF_NAME} = *-rc[0-9]* ]]; then
             echo "CHANNEL=test" >> "$GITHUB_ENV"
           fi
+      - name: Set CHANNEL Release (for release)
+        if: ${{ (github.event_name == 'pull_request' && startsWith(github.base_ref, 'release')) || startsWith(github.ref, 'release') }}
+        run: |
+          echo "CHANNEL=test" >> "$GITHUB_ENV"
       - name: Install conda-build and purge previous artifacts
         shell: arch -arch arm64 bash {0}
         run: |

.github/workflows/test-m1.yml

@@ -7,7 +7,10 @@ on:
     branches:
       - nightly
       - main
+      - release/*
   workflow_dispatch:
+env:
+  CHANNEL: "nightly"
 jobs:
   tests:
     name: "Unit-tests on M1"
@@ -19,6 +22,10 @@ jobs:
     steps:
       - name: Checkout repository
         uses: actions/checkout@v2
+      - name: Set Release CHANNEL (for release)
+        if: ${{ (github.event_name == 'pull_request' && startsWith(github.base_ref, 'release')) || startsWith(github.ref, 'release') }}
+        run: |
+          echo "CHANNEL=test" >> "$GITHUB_ENV"
       - name: Install TorchVision
         shell: arch -arch arm64 bash {0}
         env:
@@ -30,7 +37,11 @@ jobs:
           export PATH=~/miniconda3/bin:$PATH
           set -ex
           conda create -yp ${ENV_NAME} python=${PY_VERS} numpy libpng jpeg scipy
+<<<<<<< HEAD
           conda run -p ${ENV_NAME} python3 -mpip install torch --extra-index-url=https://download.pytorch.org/whl/nightly
+=======
+          conda run -p ${ENV_NAME} python3 -mpip install --pre torch --extra-index-url=https://download.pytorch.org/whl/${CHANNEL}
+>>>>>>> 369317f45 (Make sure we are building against test channel for release (#6168))
           conda run -p ${ENV_NAME} python3 setup.py develop
           conda run -p ${ENV_NAME} python3 -mpip install pytest pytest-mock av
       - name: Run tests