|
15 | 15 |
|
16 | 16 | jobs:
|
17 | 17 | generate-matrix:
|
18 |
| - uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@release/2.3 |
| 18 | + uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main |
19 | 19 | with:
|
20 | 20 | package-type: wheel
|
21 | 21 | os: linux
|
22 | 22 | test-infra-repository: pytorch/test-infra
|
23 | 23 | test-infra-ref: main
|
24 |
| - channel: test |
25 | 24 | with-rocm: false
|
26 | 25 | with-cpu: false
|
27 | 26 |
|
|
41 | 40 | smoke-test-script: packaging/smoke_test_script.sh
|
42 | 41 | package-name: torch_tensorrt
|
43 | 42 | name: Build torch-tensorrt whl package
|
44 |
| - uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@release/2.3 |
| 43 | + uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main |
45 | 44 | with:
|
46 | 45 | repository: ${{ matrix.repository }}
|
47 | 46 | ref: ""
|
|
66 | 65 | package-name: torch_tensorrt
|
67 | 66 | pre-script: packaging/pre_build_script.sh
|
68 | 67 | post-script: packaging/post_build_script.sh
|
69 |
| - uses: pytorch/tensorrt/.github/workflows/linux-test.yml@release/2.3 |
| 68 | + uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main |
70 | 69 | with:
|
71 | 70 | job-name: tests-py-torchscript-fe
|
72 | 71 | repository: "pytorch/tensorrt"
|
@@ -104,7 +103,7 @@ jobs:
|
104 | 103 | package-name: torch_tensorrt
|
105 | 104 | pre-script: packaging/pre_build_script.sh
|
106 | 105 | post-script: packaging/post_build_script.sh
|
107 |
| - uses: pytorch/tensorrt/.github/workflows/linux-test.yml@release/2.3 |
| 106 | + uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main |
108 | 107 | with:
|
109 | 108 | job-name: tests-py-dynamo-converters
|
110 | 109 | repository: "pytorch/tensorrt"
|
@@ -133,7 +132,7 @@ jobs:
|
133 | 132 | package-name: torch_tensorrt
|
134 | 133 | pre-script: packaging/pre_build_script.sh
|
135 | 134 | post-script: packaging/post_build_script.sh
|
136 |
| - uses: pytorch/tensorrt/.github/workflows/linux-test.yml@release/2.3 |
| 135 | + uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main |
137 | 136 | with:
|
138 | 137 | job-name: tests-py-dynamo-fe
|
139 | 138 | repository: "pytorch/tensorrt"
|
@@ -163,7 +162,7 @@ jobs:
|
163 | 162 | package-name: torch_tensorrt
|
164 | 163 | pre-script: packaging/pre_build_script.sh
|
165 | 164 | post-script: packaging/post_build_script.sh
|
166 |
| - uses: pytorch/tensorrt/.github/workflows/linux-test.yml@release/2.3 |
| 165 | + uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main |
167 | 166 | with:
|
168 | 167 | job-name: tests-py-dynamo-serde
|
169 | 168 | repository: "pytorch/tensorrt"
|
@@ -192,7 +191,7 @@ jobs:
|
192 | 191 | package-name: torch_tensorrt
|
193 | 192 | pre-script: packaging/pre_build_script.sh
|
194 | 193 | post-script: packaging/post_build_script.sh
|
195 |
| - uses: pytorch/tensorrt/.github/workflows/linux-test.yml@release/2.3 |
| 194 | + uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main |
196 | 195 | with:
|
197 | 196 | job-name: tests-py-torch-compile-be
|
198 | 197 | repository: "pytorch/tensorrt"
|
@@ -223,7 +222,7 @@ jobs:
|
223 | 222 | package-name: torch_tensorrt
|
224 | 223 | pre-script: packaging/pre_build_script.sh
|
225 | 224 | post-script: packaging/post_build_script.sh
|
226 |
| - uses: pytorch/tensorrt/.github/workflows/linux-test.yml@release/2.3 |
| 225 | + uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main |
227 | 226 | with:
|
228 | 227 | job-name: tests-py-dynamo-core
|
229 | 228 | repository: "pytorch/tensorrt"
|
|
0 commit comments