@@ -315,39 +315,6 @@ jobs:
315315 python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/l1_ts_models_tests_results.xml -n auto models/
316316 popd
317317
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-
351318 L2-torch-compile-tests :
352319 name : L2 torch compile tests
353320 needs : [filter-matrix, build, L1-torch-compile-tests, L1-dynamo-compile-tests, L1-dynamo-core-tests, L1-torchscript-tests]
@@ -494,6 +461,39 @@ jobs:
494461 python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/l2_ts_integrations_tests_results.xml -n auto integrations/
495462 popd
496463
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}/dynamo_distributed_test_results.xml distributed/test_nccl_ops.py
495+ popd
496+
497497concurrency :
498498 group : ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-tensorrt-${{ inputs.repository }}-${{ github.event_name == 'workflow_dispatch' }}-${{ inputs.job-name }}
499499 cancel-in-progress : true
0 commit comments