@@ -315,9 +315,9 @@ jobs:
315315 python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/l1_ts_models_tests_results.xml -n auto models/
316316 popd
317317
318- L2-torch-compile -tests :
319- name : L2 torch compile tests
320- needs : [filter-matrix, build, L1-torch-compile -tests, L1 -dynamo-compile -tests, L1-dynamo -core-tests, L1 -torchscript-tests]
318+ L1-dynamo-distributed -tests :
319+ name : L1-dynamo-distributed- tests
320+ needs : [filter-matrix, build,L0-dynamo-core -tests, L0 -dynamo-converter -tests, L0-py -core-tests, L0 -torchscript-tests]
321321 strategy :
322322 fail-fast : false
323323 matrix :
@@ -329,7 +329,7 @@ jobs:
329329 smoke-test-script : packaging/smoke_test_script.sh
330330 uses : ./.github/workflows/linux-test.yml
331331 with :
332- job-name : L2-torch-compile -tests
332+ job-name : L1-dynamo-distributed -tests
333333 repository : " pytorch/tensorrt"
334334 ref : " "
335335 test-infra-repository : pytorch/test-infra
@@ -338,15 +338,19 @@ jobs:
338338 pre-script : ${{ matrix.pre-script }}
339339 script : |
340340 set -euo pipefail
341+ export USE_HOST_DEPS=1
342+ export CI_BUILD=1
343+ export USE_TRTLLM_PLUGINS=1
344+ dnf install -y mpich mpich-devel openmpi openmpi-devel
341345 pushd .
342- cd tests/py/dynamo/
343- python -m pytest -m "not critical" -ra -n auto --junitxml=${RUNNER_TEST_RESULTS_DIR}/l2_torch_compile_models_tests_results.xml --ir torch_compile models/test_models.py
344- python -m pytest -m "not critical" - ra -n auto -- junitxml=${RUNNER_TEST_RESULTS_DIR}/l2_torch_compile_dyn_models_tests_results .xml --ir torch_compile models/test_dyn_models .py
346+ cd tests/py
347+ cd dynamo
348+ python -m pytest -ra -- junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_distributed_test_results .xml distributed/test_nccl_ops .py
345349 popd
346350
347- L1-dynamo-distributed -tests :
348- name : Test dynamo distributed [Python]
349- needs : [filter-matrix, build]
351+ L2-torch-compile -tests :
352+ name : L2 torch compile tests
353+ needs : [filter-matrix, build, L1-torch-compile-tests, L1-dynamo-compile-tests, L1-dynamo-core-tests, L1-torchscript-tests ]
350354 strategy :
351355 fail-fast : false
352356 matrix :
@@ -358,7 +362,7 @@ jobs:
358362 smoke-test-script : packaging/smoke_test_script.sh
359363 uses : ./.github/workflows/linux-test.yml
360364 with :
361- job-name : tests-py-dynamo-distributed
365+ job-name : L2-torch-compile-tests
362366 repository : " pytorch/tensorrt"
363367 ref : " "
364368 test-infra-repository : pytorch/test-infra
@@ -367,17 +371,12 @@ jobs:
367371 pre-script : ${{ matrix.pre-script }}
368372 script : |
369373 set -euo pipefail
370- export USE_HOST_DEPS=1
371- export CI_BUILD=1
372- export USE_TRTLLM_PLUGINS=1
373- dnf install -y mpich mpich-devel openmpi openmpi-devel
374374 pushd .
375- cd tests/py
376- cd dynamo
377- python -m pytest -ra -- junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_distributed_test_results .xml distributed/test_nccl_ops .py
375+ cd tests/py/dynamo/
376+ python -m pytest -m "not critical" -ra -n auto --junitxml=${RUNNER_TEST_RESULTS_DIR}/l2_torch_compile_models_tests_results.xml --ir torch_compile models/test_models.py
377+ python -m pytest -m "not critical" - ra -n auto -- junitxml=${RUNNER_TEST_RESULTS_DIR}/l2_torch_compile_dyn_models_tests_results .xml --ir torch_compile models/test_dyn_models .py
378378 popd
379379
380-
381380 L2-dynamo-compile-tests :
382381 name : L2 dynamo compile tests
383382 needs : [filter-matrix, build, L1-dynamo-compile-tests, L1-dynamo-core-tests, L1-torch-compile-tests, L1-torchscript-tests]
0 commit comments