diff --git a/.github/workflows/build_docs.yml b/.github/workflows/build_docs.yml
index 5da61c3b28..41d3d797e1 100644
--- a/.github/workflows/build_docs.yml
+++ b/.github/workflows/build_docs.yml
@@ -22,6 +22,7 @@ jobs:
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@release/2.1
     with:
+      test-infra-ref: release/2.1
       job-name: Build doc
       runner: linux.g5.4xlarge.nvidia.gpu
       repository: pytorch/audio
diff --git a/.github/workflows/lint.yml b/.github/workflows/lint.yml
index bce6eb1140..0b05e1e970 100644
--- a/.github/workflows/lint.yml
+++ b/.github/workflows/lint.yml
@@ -13,6 +13,7 @@ jobs:
   python-source-and-configs:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@release/2.1
     with:
+      test-infra-ref: release/2.1
       repository: pytorch/audio
       script: |
         set -euo pipefail
diff --git a/.github/workflows/unittest-linux-cpu.yml b/.github/workflows/unittest-linux-cpu.yml
index 2fdadc923f..4a955f1c64 100644
--- a/.github/workflows/unittest-linux-cpu.yml
+++ b/.github/workflows/unittest-linux-cpu.yml
@@ -17,6 +17,7 @@ jobs:
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@release/2.1
     with:
+      test-infra-ref: release/2.1
       runner: linux.12xlarge
       repository: pytorch/audio
       timeout: 120
diff --git a/.github/workflows/unittest-linux-gpu.yml b/.github/workflows/unittest-linux-gpu.yml
index 9cfac4eed3..9fcfc8dd8f 100644
--- a/.github/workflows/unittest-linux-gpu.yml
+++ b/.github/workflows/unittest-linux-gpu.yml
@@ -18,6 +18,7 @@ jobs:
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@release/2.1
     with:
+      test-infra-ref: release/2.1
       runner: linux.g5.4xlarge.nvidia.gpu
       repository: pytorch/audio
       gpu-arch-type: cuda
diff --git a/.github/workflows/unittest-macos-cpu.yml b/.github/workflows/unittest-macos-cpu.yml
index 3e846c82c9..bea5f8743b 100644
--- a/.github/workflows/unittest-macos-cpu.yml
+++ b/.github/workflows/unittest-macos-cpu.yml
@@ -16,6 +16,7 @@ jobs:
   tests:
     uses: pytorch/test-infra/.github/workflows/macos_job.yml@release/2.1
     with:
+      test-infra-ref: release/2.1
       runner: macos-12
       repository: pytorch/audio
       timeout: 180
diff --git a/.github/workflows/unittest-windows-cpu.yml b/.github/workflows/unittest-windows-cpu.yml
index 7a0e3761a8..ebe390188d 100644
--- a/.github/workflows/unittest-windows-cpu.yml
+++ b/.github/workflows/unittest-windows-cpu.yml
@@ -13,6 +13,7 @@ jobs:
   unittests-windows-cpu:
     uses: pytorch/test-infra/.github/workflows/windows_job.yml@release/2.1
     with:
+      test-infra-ref: release/2.1
       repository: pytorch/audio
       runner: windows.4xlarge
       timeout: 180
diff --git a/.github/workflows/unittest-windows-gpu.yml b/.github/workflows/unittest-windows-gpu.yml
index 1a3febc162..a8a57ae6c7 100644
--- a/.github/workflows/unittest-windows-gpu.yml
+++ b/.github/workflows/unittest-windows-gpu.yml
@@ -13,6 +13,7 @@ jobs:
   unittests-windows-gpu:
     uses: pytorch/test-infra/.github/workflows/windows_job.yml@release/2.1
     with:
+      test-infra-ref: release/2.1
       repository: pytorch/audio
       runner: windows.g5.4xlarge.nvidia.gpu
       timeout: 360