Point test infra to fork

Author: gs-olive

.github/workflows/build-test.yml

@@ -19,7 +19,7 @@ jobs:
     with:
       package-type: wheel
       os: linux
-      test-infra-repository: pytorch/test-infra
+      test-infra-repository: gs-olive/test-infra
       test-infra-ref: release/2.2
       channel: test
       with-rocm: false
@@ -38,11 +38,11 @@ jobs:
             smoke-test-script: ""
             package-name: torch_tensorrt
     name: Build torch-tensorrt whl package
-    uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@release/2.2
+    uses: gs-olive/test-infra/.github/workflows/build_wheels_linux.yml@release/2.2
     with:
       repository: ${{ matrix.repository }}
       ref: ""
-      test-infra-repository: pytorch/test-infra
+      test-infra-repository: gs-olive/test-infra
       test-infra-ref: release/2.2
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
       pre-script: ${{ matrix.pre-script }}
@@ -71,7 +71,7 @@ jobs:
       job-name: tests-py-torchscript-fe
       repository: "pytorch/tensorrt"
       ref: ""
-      test-infra-repository: pytorch/test-infra
+      test-infra-repository: gs-olive/test-infra
       test-infra-ref: release/2.2
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
       pre-script: ${{ matrix.pre-script }}
@@ -107,7 +107,7 @@ jobs:
       job-name: tests-py-dynamo-converters
       repository: "pytorch/tensorrt"
       ref: ""
-      test-infra-repository: pytorch/test-infra
+      test-infra-repository: gs-olive/test-infra
       test-infra-ref: release/2.2
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
       pre-script: ${{ matrix.pre-script }}
@@ -135,7 +135,7 @@ jobs:
       job-name: tests-py-dynamo-fe
       repository: "pytorch/tensorrt"
       ref: ""
-      test-infra-repository: pytorch/test-infra
+      test-infra-repository: gs-olive/test-infra
       test-infra-ref: release/2.2
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
       pre-script: ${{ matrix.pre-script }}
@@ -165,7 +165,7 @@ jobs:
       job-name: tests-py-torch-compile-be
       repository: "pytorch/tensorrt"
       ref: ""
-      test-infra-repository: pytorch/test-infra
+      test-infra-repository: gs-olive/test-infra
       test-infra-ref: release/2.2
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
       pre-script: ${{ matrix.pre-script }}
@@ -194,7 +194,7 @@ jobs:
       job-name: tests-py-dynamo-core
       repository: "pytorch/tensorrt"
       ref: ""
-      test-infra-repository: pytorch/test-infra
+      test-infra-repository: gs-olive/test-infra
       test-infra-ref: release/2.2
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
       pre-script: ${{ matrix.pre-script }}

.github/workflows/linux-test.yml

@@ -21,7 +21,7 @@ on:
         type: string
       test-infra-repository:
         description: "Test infra repository to use"
-        default: "pytorch/test-infra"
+        default: "gs-olive/test-infra"
         type: string
       test-infra-ref:
         description: "Test infra reference to use"