Skip to content

Commit e9af038

Browse files
lanluo-nvidialanluo-nvidia
committed
add more tests
1 parent 6cc2faa commit e9af038

File tree

1 file changed

+185
-2
lines changed

1 file changed

+185
-2
lines changed

‎.github/workflows/build-test-tensorrt-linux.yml

Lines changed: 185 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
name: Build and Test Torch-TensorRT on Linux with Future TensorRT Versions
22

33
on:
4+
pull_request:
45
workflow_dispatch:
56

67
permissions:
@@ -103,6 +104,124 @@ jobs:
103104
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_integrations_test_results.xml integrations/
104105
popd
105106
107+
tests-py-dynamo-converters:
108+
name: Test dynamo converters [Python]
109+
needs: [generate-tensorrt-matrix, build]
110+
strategy:
111+
fail-fast: false
112+
matrix:
113+
include:
114+
- repository: pytorch/tensorrt
115+
package-name: torch_tensorrt
116+
pre-script: packaging/pre_build_script.sh
117+
post-script: packaging/post_build_script.sh
118+
smoke-test-script: packaging/smoke_test_script.sh
119+
uses: ./.github/workflows/linux-test.yml
120+
with:
121+
job-name: tests-py-dynamo-converters
122+
repository: "pytorch/tensorrt"
123+
ref: ""
124+
test-infra-repository: pytorch/test-infra
125+
test-infra-ref: main
126+
build-matrix: ${{ needs.generate-tensorrt-matrix.outputs.matrix }}
127+
pre-script: ${{ matrix.pre-script }}
128+
script: |
129+
export USE_HOST_DEPS=1
130+
export CI_BUILD=1
131+
pushd .
132+
cd tests/py/dynamo
133+
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml -n 8 conversion/
134+
popd
135+
136+
tests-py-dynamo-fe:
137+
name: Test dynamo frontend [Python]
138+
needs: [generate-tensorrt-matrix, build]
139+
strategy:
140+
fail-fast: false
141+
matrix:
142+
include:
143+
- repository: pytorch/tensorrt
144+
package-name: torch_tensorrt
145+
pre-script: packaging/pre_build_script.sh
146+
post-script: packaging/post_build_script.sh
147+
smoke-test-script: packaging/smoke_test_script.sh
148+
uses: ./.github/workflows/linux-test.yml
149+
with:
150+
job-name: tests-py-dynamo-fe
151+
repository: "pytorch/tensorrt"
152+
ref: ""
153+
test-infra-repository: pytorch/test-infra
154+
test-infra-ref: main
155+
build-matrix: ${{ needs.generate-tensorrt-matrix.outputs.matrix }}
156+
pre-script: ${{ matrix.pre-script }}
157+
script: |
158+
export USE_HOST_DEPS=1
159+
export CI_BUILD=1
160+
pushd .
161+
cd tests/py/dynamo
162+
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dyn_models_export.xml --ir dynamo models/
163+
popd
164+
165+
tests-py-dynamo-serde:
166+
name: Test dynamo export serde [Python]
167+
needs: [generate-tensorrt-matrix, build]
168+
strategy:
169+
fail-fast: false
170+
matrix:
171+
include:
172+
- repository: pytorch/tensorrt
173+
package-name: torch_tensorrt
174+
pre-script: packaging/pre_build_script.sh
175+
post-script: packaging/post_build_script.sh
176+
smoke-test-script: packaging/smoke_test_script.sh
177+
uses: ./.github/workflows/linux-test.yml
178+
with:
179+
job-name: tests-py-dynamo-serde
180+
repository: "pytorch/tensorrt"
181+
ref: ""
182+
test-infra-repository: pytorch/test-infra
183+
test-infra-ref: main
184+
build-matrix: ${{ needs.generate-tensorrt-matrix.outputs.matrix }}
185+
pre-script: ${{ matrix.pre-script }}
186+
script: |
187+
export USE_HOST_DEPS=1
188+
export CI_BUILD=1
189+
pushd .
190+
cd tests/py/dynamo
191+
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/export_serde_test_results.xml --ir dynamo models/test_export_serde.py
192+
popd
193+
194+
tests-py-torch-compile-be:
195+
name: Test torch compile backend [Python]
196+
needs: [generate-tensorrt-matrix, build]
197+
strategy:
198+
fail-fast: false
199+
matrix:
200+
include:
201+
- repository: pytorch/tensorrt
202+
package-name: torch_tensorrt
203+
pre-script: packaging/pre_build_script.sh
204+
post-script: packaging/post_build_script.sh
205+
smoke-test-script: packaging/smoke_test_script.sh
206+
uses: ./.github/workflows/linux-test.yml
207+
with:
208+
job-name: tests-py-torch-compile-be
209+
repository: "pytorch/tensorrt"
210+
ref: ""
211+
test-infra-repository: pytorch/test-infra
212+
test-infra-ref: main
213+
build-matrix: ${{ needs.generate-tensorrt-matrix.outputs.matrix }}
214+
pre-script: ${{ matrix.pre-script }}
215+
script: |
216+
export USE_HOST_DEPS=1
217+
export CI_BUILD=1
218+
pushd .
219+
cd tests/py/dynamo
220+
python -m pytest -ra -n 10 --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_be_test_results.xml backend/
221+
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
222+
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_dyn_models_export.xml --ir torch_compile models/test_dyn_models.py
223+
popd
224+
106225
tests-py-dynamo-core:
107226
name: Test dynamo core [Python]
108227
needs: [generate-tensorrt-matrix, build]
@@ -129,7 +248,71 @@ jobs:
129248
export CI_BUILD=1
130249
pushd .
131250
cd tests/py/dynamo
132-
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_runtime_test_results.xml runtime/test_000_*
251+
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/
133252
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_partitioning_test_results.xml partitioning/
134253
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_lowering_test_results.xml lowering/
135-
popd
254+
popd
255+
256+
tests-py-dynamo-cudagraphs:
257+
name: Test dynamo cudagraphs [Python]
258+
needs: [generate-tensorrt-matrix, build]
259+
strategy:
260+
fail-fast: false
261+
matrix:
262+
include:
263+
- repository: pytorch/tensorrt
264+
package-name: torch_tensorrt
265+
pre-script: packaging/pre_build_script.sh
266+
post-script: packaging/post_build_script.sh
267+
smoke-test-script: packaging/smoke_test_script.sh
268+
uses: ./.github/workflows/linux-test.yml
269+
with:
270+
job-name: tests-py-dynamo-cudagraphs
271+
repository: "pytorch/tensorrt"
272+
ref: ""
273+
test-infra-repository: pytorch/test-infra
274+
test-infra-ref: main
275+
build-matrix: ${{ needs.generate-tensorrt-matrix.outputs.matrix }}
276+
pre-script: ${{ matrix.pre-script }}
277+
script: |
278+
export USE_HOST_DEPS=1
279+
export CI_BUILD=1
280+
pushd .
281+
cd tests/py/dynamo
282+
nvidia-smi
283+
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
284+
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
285+
popd
286+
287+
tests-py-core:
288+
name: Test core [Python]
289+
needs: [generate-tensorrt-matrix, build]
290+
strategy:
291+
fail-fast: false
292+
matrix:
293+
include:
294+
- repository: pytorch/tensorrt
295+
package-name: torch_tensorrt
296+
pre-script: packaging/pre_build_script.sh
297+
post-script: packaging/post_build_script.sh
298+
smoke-test-script: packaging/smoke_test_script.sh
299+
uses: ./.github/workflows/linux-test.yml
300+
with:
301+
job-name: tests-py-core
302+
repository: "pytorch/tensorrt"
303+
ref: ""
304+
test-infra-repository: pytorch/test-infra
305+
test-infra-ref: main
306+
build-matrix: ${{ needs.generate-tensorrt-matrix.outputs.matrix }}
307+
pre-script: ${{ matrix.pre-script }}
308+
script: |
309+
export USE_HOST_DEPS=1
310+
export CI_BUILD=1
311+
pushd .
312+
cd tests/py/core
313+
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_core_test_results.xml .
314+
popd
315+
316+
concurrency:
317+
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-${{ inputs.repository }}-${{ github.event_name == 'workflow_dispatch' }}-${{ inputs.job-name }}
318+
cancel-in-progress: true

0 commit comments

Comments
 (0)