6262 post-script : packaging/post_build_script.sh
6363 smoke-test-script : packaging/smoke_test_script.sh
6464 package-name : torch_tensorrt
65- name : Build torch-tensorrt whl package for SBSA
65+ name : Build SBSA torch-tensorrt whl package
6666 uses : ./.github/workflows/build_linux.yml
6767 with :
6868 repository : ${{ matrix.repository }}
@@ -78,284 +78,6 @@ jobs:
7878 trigger-event : ${{ github.event_name }}
7979 architecture : " aarch64"
8080
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-
35981concurrency :
36082 group : ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-${{ inputs.repository }}-${{ github.event_name == 'workflow_dispatch' }}-${{ inputs.job-name }}
36183 cancel-in-progress : true
0 commit comments