|
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" |
|