@@ -315,9 +315,9 @@ jobs:
315
315
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/l1_ts_models_tests_results.xml -n auto models/
316
316
popd
317
317
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 : Test dynamo distributed [Python]
320
+ needs : [filter-matrix, build,L0-dynamo-core -tests, L0 -dynamo-converter -tests, L0-py -core-tests, L0 -torchscript-tests]
321
321
strategy :
322
322
fail-fast : false
323
323
matrix :
@@ -329,7 +329,7 @@ jobs:
329
329
smoke-test-script : packaging/smoke_test_script.sh
330
330
uses : ./.github/workflows/linux-test.yml
331
331
with :
332
- job-name : L2-torch-compile-tests
332
+ job-name : tests-py-dynamo-distributed
333
333
repository : " pytorch/tensorrt"
334
334
ref : " "
335
335
test-infra-repository : pytorch/test-infra
@@ -338,15 +338,19 @@ jobs:
338
338
pre-script : ${{ matrix.pre-script }}
339
339
script : |
340
340
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
341
345
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
345
349
popd
346
350
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 ]
350
354
strategy :
351
355
fail-fast : false
352
356
matrix :
@@ -358,7 +362,7 @@ jobs:
358
362
smoke-test-script : packaging/smoke_test_script.sh
359
363
uses : ./.github/workflows/linux-test.yml
360
364
with :
361
- job-name : tests-py-dynamo-distributed
365
+ job-name : L2-torch-compile-tests
362
366
repository : " pytorch/tensorrt"
363
367
ref : " "
364
368
test-infra-repository : pytorch/test-infra
@@ -367,17 +371,12 @@ jobs:
367
371
pre-script : ${{ matrix.pre-script }}
368
372
script : |
369
373
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
374
374
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
378
378
popd
379
379
380
-
381
380
L2-dynamo-compile-tests :
382
381
name : L2 dynamo compile tests
383
382
needs : [filter-matrix, build, L1-dynamo-compile-tests, L1-dynamo-core-tests, L1-torch-compile-tests, L1-torchscript-tests]
0 commit comments