Skip to content

Commit 8496611

Browse files
committed
changing location of the L1 distributed tests
1 parent 226ed04 commit 8496611

File tree

1 file changed

+18
-19
lines changed

1 file changed

+18
-19
lines changed

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

Lines changed: 18 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -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: Test dynamo distributed [Python]
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: tests-py-dynamo-distributed
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

Comments
 (0)