From e6accb0c70aff448415da620770ea083a9786c9a Mon Sep 17 00:00:00 2001
From: gs-olive <113141689+gs-olive@users.noreply.github.com>
Date: Fri, 26 Apr 2024 13:48:30 -0700
Subject: [PATCH] fix: Point infra branch to main

---
 .github/workflows/build-test.yml | 17 ++++++++---------
 1 file changed, 8 insertions(+), 9 deletions(-)

diff --git a/.github/workflows/build-test.yml b/.github/workflows/build-test.yml
index f4d39bd056..c52b98b202 100644
--- a/.github/workflows/build-test.yml
+++ b/.github/workflows/build-test.yml
@@ -15,13 +15,12 @@ on:
 
 jobs:
   generate-matrix:
-    uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@release/2.3
+    uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
     with:
       package-type: wheel
       os: linux
       test-infra-repository: pytorch/test-infra
       test-infra-ref: main
-      channel: test
       with-rocm: false
       with-cpu: false
 
@@ -41,7 +40,7 @@ jobs:
             smoke-test-script: packaging/smoke_test_script.sh
             package-name: torch_tensorrt
     name: Build torch-tensorrt whl package
-    uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@release/2.3
+    uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
     with:
       repository: ${{ matrix.repository }}
       ref: ""
@@ -66,7 +65,7 @@ jobs:
             package-name: torch_tensorrt
             pre-script: packaging/pre_build_script.sh
             post-script: packaging/post_build_script.sh
-    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@release/2.3
+    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
     with:
       job-name: tests-py-torchscript-fe
       repository: "pytorch/tensorrt"
@@ -104,7 +103,7 @@ jobs:
             package-name: torch_tensorrt
             pre-script: packaging/pre_build_script.sh
             post-script: packaging/post_build_script.sh
-    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@release/2.3
+    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
     with:
       job-name: tests-py-dynamo-converters
       repository: "pytorch/tensorrt"
@@ -133,7 +132,7 @@ jobs:
             package-name: torch_tensorrt
             pre-script: packaging/pre_build_script.sh
             post-script: packaging/post_build_script.sh
-    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@release/2.3
+    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
     with:
       job-name: tests-py-dynamo-fe
       repository: "pytorch/tensorrt"
@@ -163,7 +162,7 @@ jobs:
             package-name: torch_tensorrt
             pre-script: packaging/pre_build_script.sh
             post-script: packaging/post_build_script.sh
-    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@release/2.3
+    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
     with:
       job-name: tests-py-dynamo-serde
       repository: "pytorch/tensorrt"
@@ -192,7 +191,7 @@ jobs:
             package-name: torch_tensorrt
             pre-script: packaging/pre_build_script.sh
             post-script: packaging/post_build_script.sh
-    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@release/2.3
+    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
     with:
       job-name: tests-py-torch-compile-be
       repository: "pytorch/tensorrt"
@@ -223,7 +222,7 @@ jobs:
             package-name: torch_tensorrt
             pre-script: packaging/pre_build_script.sh
             post-script: packaging/post_build_script.sh
-    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@release/2.3
+    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
     with:
       job-name: tests-py-dynamo-core
       repository: "pytorch/tensorrt"