Skip to content

Commit 24264e5

Browse files
committed
moving tests to L2
1 parent 2f2cd31 commit 24264e5

File tree

1 file changed

+33
-33
lines changed

1 file changed

+33
-33
lines changed

.github/workflows/build-test-linux-x86_64.yml

Lines changed: 33 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -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}/l2_dynamo_distributed_test_results.xml distributed/test_nccl_ops.py
495+
popd
496+
497497
concurrency:
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

Comments
 (0)