@@ -315,39 +315,6 @@ 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
- 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]
321
- strategy :
322
- fail-fast : false
323
- matrix :
324
- include :
325
- - repository : pytorch/tensorrt
326
- package-name : torch_tensorrt
327
- pre-script : packaging/pre_build_script.sh
328
- post-script : packaging/post_build_script.sh
329
- smoke-test-script : packaging/smoke_test_script.sh
330
- uses : ./.github/workflows/linux-test.yml
331
- with :
332
- job-name : L1-dynamo-distributed-tests
333
- repository : " pytorch/tensorrt"
334
- ref : " "
335
- test-infra-repository : pytorch/test-infra
336
- test-infra-ref : main
337
- build-matrix : ${{ needs.filter-matrix.outputs.matrix }}
338
- pre-script : ${{ matrix.pre-script }}
339
- script : |
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
345
- pushd .
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
349
- popd
350
-
351
318
L2-torch-compile-tests :
352
319
name : L2 torch compile tests
353
320
needs : [filter-matrix, build, L1-torch-compile-tests, L1-dynamo-compile-tests, L1-dynamo-core-tests, L1-torchscript-tests]
@@ -494,6 +461,39 @@ jobs:
494
461
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/l2_ts_integrations_tests_results.xml -n auto integrations/
495
462
popd
496
463
464
+ L2-dynamo-distributed-tests :
465
+ name : L2 dynamo distributed tests
466
+ needs : [filter-matrix, build, L1-dynamo-core-tests, L1-dynamo-compile-tests, L1-torch-compile-tests, L1-torchscript-tests]
467
+ strategy :
468
+ fail-fast : false
469
+ matrix :
470
+ include :
471
+ - repository : pytorch/tensorrt
472
+ package-name : torch_tensorrt
473
+ pre-script : packaging/pre_build_script.sh
474
+ post-script : packaging/post_build_script.sh
475
+ smoke-test-script : packaging/smoke_test_script.sh
476
+ uses : ./.github/workflows/linux-test.yml
477
+ with :
478
+ job-name : L2-dynamo-distributed-tests
479
+ repository : " pytorch/tensorrt"
480
+ ref : " "
481
+ test-infra-repository : pytorch/test-infra
482
+ test-infra-ref : main
483
+ build-matrix : ${{ needs.filter-matrix.outputs.matrix }}
484
+ pre-script : ${{ matrix.pre-script }}
485
+ script : |
486
+ set -euo pipefail
487
+ export USE_HOST_DEPS=1
488
+ export CI_BUILD=1
489
+ export USE_TRTLLM_PLUGINS=1
490
+ dnf install -y mpich mpich-devel openmpi openmpi-devel
491
+ pushd .
492
+ cd tests/py
493
+ cd dynamo
494
+ python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/l2_dynamo_distributed_test_results.xml distributed/test_nccl_ops.py
495
+ popd
496
+
497
497
concurrency :
498
498
group : ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-tensorrt-${{ inputs.repository }}-${{ github.event_name == 'workflow_dispatch' }}-${{ inputs.job-name }}
499
499
cancel-in-progress : true
0 commit comments