62
62
post-script : packaging/post_build_script.sh
63
63
smoke-test-script : packaging/smoke_test_script.sh
64
64
package-name : torch_tensorrt
65
- name : Build torch-tensorrt whl package for SBSA
65
+ name : Build SBSA torch-tensorrt whl package
66
66
uses : ./.github/workflows/build_linux.yml
67
67
with :
68
68
repository : ${{ matrix.repository }}
@@ -78,284 +78,6 @@ jobs:
78
78
trigger-event : ${{ github.event_name }}
79
79
architecture : " aarch64"
80
80
81
- tests-py-torchscript-fe :
82
- name : Test torchscript frontend [Python]
83
- needs : [filter-matrix, build]
84
- if : false
85
- strategy :
86
- fail-fast : false
87
- matrix :
88
- include :
89
- - repository : pytorch/tensorrt
90
- package-name : torch_tensorrt
91
- pre-script : packaging/pre_build_script.sh
92
- post-script : packaging/post_build_script.sh
93
- smoke-test-script : packaging/smoke_test_script.sh
94
- uses : ./.github/workflows/linux-test.yml
95
- with :
96
- job-name : tests-py-torchscript-fe
97
- repository : " pytorch/tensorrt"
98
- ref : " "
99
- test-infra-repository : pytorch/test-infra
100
- test-infra-ref : main
101
- build-matrix : ${{ needs.filter-matrix.outputs.matrix }}
102
- pre-script : ${{ matrix.pre-script }}
103
- architecture : " aarch64"
104
- script : |
105
- set -euo pipefail
106
- export USE_HOST_DEPS=1
107
- export CI_BUILD=1
108
- export LD_LIBRARY_PATH=/usr/lib64:$LD_LIBRARY_PATH
109
- pushd .
110
- cd tests/modules
111
- python hub.py
112
- popd
113
- pushd .
114
- cd tests/py/ts
115
- python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_api_test_results.xml api/
116
- python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_models_test_results.xml models/
117
- python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_integrations_test_results.xml integrations/
118
- popd
119
-
120
- tests-py-dynamo-converters :
121
- name : Test dynamo converters [Python]
122
- needs : [filter-matrix, build]
123
- if : false
124
- strategy :
125
- fail-fast : false
126
- matrix :
127
- include :
128
- - repository : pytorch/tensorrt
129
- package-name : torch_tensorrt
130
- pre-script : packaging/pre_build_script.sh
131
- post-script : packaging/post_build_script.sh
132
- smoke-test-script : packaging/smoke_test_script.sh
133
- uses : ./.github/workflows/linux-test.yml
134
- with :
135
- job-name : tests-py-dynamo-converters
136
- repository : " pytorch/tensorrt"
137
- ref : " "
138
- test-infra-repository : pytorch/test-infra
139
- test-infra-ref : main
140
- build-matrix : ${{ needs.filter-matrix.outputs.matrix }}
141
- pre-script : ${{ matrix.pre-script }}
142
- architecture : " aarch64"
143
- script : |
144
- set -euo pipefail
145
- export USE_HOST_DEPS=1
146
- export CI_BUILD=1
147
- pushd .
148
- cd tests/py
149
- cd dynamo
150
- python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml -n 4 conversion/
151
- python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml automatic_plugin/test_automatic_plugin.py
152
- python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml automatic_plugin/test_automatic_plugin_with_attrs.py
153
- popd
154
-
155
- tests-py-dynamo-fe :
156
- name : Test dynamo frontend [Python]
157
- needs : [filter-matrix, build]
158
- if : false
159
- strategy :
160
- fail-fast : false
161
- matrix :
162
- include :
163
- - repository : pytorch/tensorrt
164
- package-name : torch_tensorrt
165
- pre-script : packaging/pre_build_script.sh
166
- post-script : packaging/post_build_script.sh
167
- smoke-test-script : packaging/smoke_test_script.sh
168
- uses : ./.github/workflows/linux-test.yml
169
- with :
170
- job-name : tests-py-dynamo-fe
171
- repository : " pytorch/tensorrt"
172
- ref : " "
173
- test-infra-repository : pytorch/test-infra
174
- test-infra-ref : main
175
- build-matrix : ${{ needs.filter-matrix.outputs.matrix }}
176
- pre-script : ${{ matrix.pre-script }}
177
- architecture : " aarch64"
178
- script : |
179
- set -euo pipefail
180
- export USE_HOST_DEPS=1
181
- export CI_BUILD=1
182
- pushd .
183
- cd tests/py
184
- cd dynamo
185
- python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dyn_models_export.xml --ir dynamo models/
186
- popd
187
-
188
- tests-py-dynamo-serde :
189
- name : Test dynamo export serde [Python]
190
- needs : [filter-matrix, build]
191
- if : false
192
- strategy :
193
- fail-fast : false
194
- matrix :
195
- include :
196
- - repository : pytorch/tensorrt
197
- package-name : torch_tensorrt
198
- pre-script : packaging/pre_build_script.sh
199
- post-script : packaging/post_build_script.sh
200
- smoke-test-script : packaging/smoke_test_script.sh
201
- uses : ./.github/workflows/linux-test.yml
202
- with :
203
- job-name : tests-py-dynamo-serde
204
- repository : " pytorch/tensorrt"
205
- ref : " "
206
- test-infra-repository : pytorch/test-infra
207
- test-infra-ref : main
208
- build-matrix : ${{ needs.filter-matrix.outputs.matrix }}
209
- pre-script : ${{ matrix.pre-script }}
210
- architecture : " aarch64"
211
- script : |
212
- set -euo pipefail
213
- export USE_HOST_DEPS=1
214
- export CI_BUILD=1
215
- pushd .
216
- cd tests/py
217
- cd dynamo
218
- python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/export_serde_test_results.xml --ir dynamo models/test_export_serde.py
219
- python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/reexport_test_results.xml --ir dynamo models/test_reexport.py
220
- popd
221
-
222
- tests-py-torch-compile-be :
223
- name : Test torch compile backend [Python]
224
- needs : [filter-matrix, build]
225
- if : false
226
- strategy :
227
- fail-fast : false
228
- matrix :
229
- include :
230
- - repository : pytorch/tensorrt
231
- package-name : torch_tensorrt
232
- pre-script : packaging/pre_build_script.sh
233
- post-script : packaging/post_build_script.sh
234
- smoke-test-script : packaging/smoke_test_script.sh
235
- uses : ./.github/workflows/linux-test.yml
236
- with :
237
- job-name : tests-py-torch-compile-be
238
- repository : " pytorch/tensorrt"
239
- ref : " "
240
- test-infra-repository : pytorch/test-infra
241
- test-infra-ref : main
242
- build-matrix : ${{ needs.filter-matrix.outputs.matrix }}
243
- pre-script : ${{ matrix.pre-script }}
244
- architecture : " aarch64"
245
- script : |
246
- set -euo pipefail
247
- export USE_HOST_DEPS=1
248
- export CI_BUILD=1
249
- pushd .
250
- cd tests/py
251
- cd dynamo
252
- python -m pytest -ra -n 10 --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_be_test_results.xml backend/
253
- python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_complete_be_e2e_test_results.xml --ir torch_compile models/test_models.py
254
- python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_dyn_models_export.xml --ir torch_compile models/test_dyn_models.py
255
- popd
256
-
257
- tests-py-dynamo-core :
258
- name : Test dynamo core [Python]
259
- needs : [filter-matrix, build]
260
- if : false
261
- strategy :
262
- fail-fast : false
263
- matrix :
264
- include :
265
- - repository : pytorch/tensorrt
266
- package-name : torch_tensorrt
267
- pre-script : packaging/pre_build_script.sh
268
- post-script : packaging/post_build_script.sh
269
- smoke-test-script : packaging/smoke_test_script.sh
270
- uses : ./.github/workflows/linux-test.yml
271
- with :
272
- job-name : tests-py-dynamo-core
273
- repository : " pytorch/tensorrt"
274
- ref : " "
275
- test-infra-repository : pytorch/test-infra
276
- test-infra-ref : main
277
- build-matrix : ${{ needs.filter-matrix.outputs.matrix }}
278
- pre-script : ${{ matrix.pre-script }}
279
- architecture : " aarch64"
280
- script : |
281
- set -euo pipefail
282
- export USE_HOST_DEPS=1
283
- export CI_BUILD=1
284
- pushd .
285
- cd tests/py
286
- cd dynamo
287
- python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_runtime_test_results.xml --ignore runtime/test_002_cudagraphs_py.py --ignore runtime/test_002_cudagraphs_cpp.py runtime/
288
- python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_partitioning_test_results.xml partitioning/
289
- python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_lowering_test_results.xml lowering/
290
- popd
291
-
292
- tests-py-dynamo-cudagraphs :
293
- name : Test dynamo cudagraphs [Python]
294
- needs : [filter-matrix, build]
295
- if : false
296
- strategy :
297
- fail-fast : false
298
- matrix :
299
- include :
300
- - repository : pytorch/tensorrt
301
- package-name : torch_tensorrt
302
- pre-script : packaging/pre_build_script.sh
303
- post-script : packaging/post_build_script.sh
304
- smoke-test-script : packaging/smoke_test_script.sh
305
- uses : ./.github/workflows/linux-test.yml
306
- with :
307
- job-name : tests-py-dynamo-cudagraphs
308
- repository : " pytorch/tensorrt"
309
- ref : " "
310
- test-infra-repository : pytorch/test-infra
311
- test-infra-ref : main
312
- build-matrix : ${{ needs.filter-matrix.outputs.matrix }}
313
- pre-script : ${{ matrix.pre-script }}
314
- architecture : " aarch64"
315
- script : |
316
- set -euo pipefail
317
- export USE_HOST_DEPS=1
318
- export CI_BUILD=1
319
- pushd .
320
- cd tests/py
321
- cd dynamo
322
- nvidia-smi
323
- python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_runtime_cudagraphs_cpp_test_results.xml runtime/test_002_cudagraphs_cpp.py || true
324
- python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_runtime_cudagraphs_py_test_results.xml runtime/test_002_cudagraphs_py.py || true
325
- popd
326
-
327
- tests-py-core :
328
- name : Test core [Python]
329
- needs : [filter-matrix, build]
330
- if : false
331
- strategy :
332
- fail-fast : false
333
- matrix :
334
- include :
335
- - repository : pytorch/tensorrt
336
- package-name : torch_tensorrt
337
- pre-script : packaging/pre_build_script.sh
338
- post-script : packaging/post_build_script.sh
339
- smoke-test-script : packaging/smoke_test_script.sh
340
- uses : ./.github/workflows/linux-test.yml
341
- with :
342
- job-name : tests-py-core
343
- repository : " pytorch/tensorrt"
344
- ref : " "
345
- test-infra-repository : pytorch/test-infra
346
- test-infra-ref : main
347
- build-matrix : ${{ needs.filter-matrix.outputs.matrix }}
348
- pre-script : ${{ matrix.pre-script }}
349
- architecture : " aarch64"
350
- script : |
351
- set -euo pipefail
352
- export USE_HOST_DEPS=1
353
- export CI_BUILD=1
354
- pushd .
355
- cd tests/py/core
356
- python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_core_test_results.xml .
357
- popd
358
-
359
81
concurrency :
360
82
group : ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-${{ inputs.repository }}-${{ github.event_name == 'workflow_dispatch' }}-${{ inputs.job-name }}
361
83
cancel-in-progress : true
0 commit comments