Switch back to pytorch as repo source

Author: gs-olive

.github/workflows/build-test-windows.yml

@@ -4,12 +4,12 @@ on: [pull_request]
 
 jobs:
   generate-matrix:
-    uses: gs-olive/test-infra/.github/workflows/generate_binary_build_matrix.yml@windows_builds
+    uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
     with:
       package-type: wheel
       os: windows
-      test-infra-repository: gs-olive/test-infra
-      test-infra-ref: windows_builds
+      test-infra-repository: pytorch/test-infra
+      test-infra-ref: main
       with-rocm: false
       with-cpu: false
 
@@ -27,12 +27,12 @@ jobs:
             smoke-test-script: packaging/smoke_test_windows.py
             package-name: torch_tensorrt
     name: Build torch-tensorrt whl package
-    uses: gs-olive/test-infra/.github/workflows/build_wheels_windows.yml@windows_builds
+    uses: pytorch/test-infra/.github/workflows/build_wheels_windows.yml@main
     with:
       repository: ${{ matrix.repository }}
       ref: ""
-      test-infra-repository: gs-olive/test-infra
-      test-infra-ref: windows_builds
+      test-infra-repository: pytorch/test-infra
+      test-infra-ref: main
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
       pre-script: ${{ matrix.pre-script }}
       smoke-test-script: ${{ matrix.smoke-test-script }}
@@ -170,8 +170,8 @@ jobs:
       job-name: tests-py-torch-compile-be
       repository: ${{ matrix.repository }}
       ref: ""
-      test-infra-repository: gs-olive/test-infra
-      test-infra-ref: windows_builds
+      test-infra-repository: pytorch/test-infra
+      test-infra-ref: main
       pre-script: packaging/pre_build_script_driver_upgrade.sh
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
       script: |