Skip to content

Don't merge #4007

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
cehongwang wants to merge 6 commits into
base: main
Choose a base branch
from
Draft

Don't merge #4007

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
4f9fe24
cherry_pick to 2.10: filter out unsupported cuda versions (#3991)
lanluo-nvidia Jan 5, 2026
5c7fa6e
test job name
lanluo-nvidia Jan 5, 2026
33d2507
fix the ci job display name issue
lanluo-nvidia Jan 5, 2026
670850b
changing the setting_to_be_engine_invariant from tuple to set
apbose Dec 31, 2025
3ef4d1d
release cut for 2.10 (#3988)
lanluo-nvidia Jan 10, 2026
9a6150f
revert print
lanluo-nvidia Jan 10, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 7 additions & 3 deletions .github/scripts/filter-matrix.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,8 @@
jetpack_cuda_versions: List[str] = ["cu126"]
# rtx 1.2 currently only supports cu129 and cu130
rtx_cuda_versions: List[str] = ["cu129", "cu130"]
# trt 10.14.1 currently only supports cu129 and cu130
trt_cuda_versions: List[str] = ["cu129", "cu130"]

jetpack_container_image: str = "nvcr.io/nvidia/l4t-jetpack:r36.4.0"
sbsa_container_image: str = "quay.io/pypa/manylinux_2_39_aarch64"
Expand Down Expand Up @@ -55,9 +57,11 @@ def filter_matrix_item(
return False
else:
if use_rtx:
if item["desired_cuda"] in rtx_cuda_versions:
return True
return False
if item["desired_cuda"] not in rtx_cuda_versions:
return False
else:
if item["desired_cuda"] not in trt_cuda_versions:
return False
if item["gpu_arch_type"] == "cuda-aarch64":
# pytorch image:pytorch/manylinuxaarch64-builder:cuda12.8 comes with glibc2.28
# however, TensorRT requires glibc2.31 on aarch64 platform
Expand Down
3 changes: 2 additions & 1 deletion .github/workflows/build-test-linux-aarch64-jetpack.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,8 @@ jobs:
post-script: packaging/post_build_script.sh
smoke-test-script: packaging/smoke_test_script.sh
package-name: torch_tensorrt
name: Build Jetpack torch-tensorrt whl package
display-name: Build Jetpack torch-tensorrt whl package
name: ${{ matrix.display-name }}
uses: ./.github/workflows/build_linux.yml
with:
repository: ${{ matrix.repository }}
Expand Down
3 changes: 2 additions & 1 deletion .github/workflows/build-test-linux-aarch64.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,8 @@ jobs:
post-script: packaging/post_build_script.sh
smoke-test-script: packaging/smoke_test_script.sh
package-name: torch_tensorrt
name: Build SBSA torch-tensorrt whl package
display-name: Build SBSA torch-tensorrt whl package
name: ${{ matrix.display-name }}
uses: ./.github/workflows/build_linux.yml
with:
repository: ${{ matrix.repository }}
Expand Down
47 changes: 31 additions & 16 deletions .github/workflows/build-test-linux-x86_64.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,8 @@ jobs:
post-script: packaging/post_build_script.sh
smoke-test-script: packaging/smoke_test_script.sh
package-name: torch_tensorrt
name: Build Linux x86_64 torch-tensorrt whl package
display-name: Build Linux x86_64 torch-tensorrt whl package
name: ${{ matrix.display-name }}
uses: ./.github/workflows/build_linux.yml
with:
repository: ${{ matrix.repository }}
Expand All @@ -80,7 +81,7 @@ jobs:
pip-install-torch-extra-args: "--extra-index-url https://pypi.org/simple"

L0-dynamo-converter-tests:
name: L0 dynamo converter tests
name: ${{ matrix.display-name }}
needs: [filter-matrix, build]
if: ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
strategy:
Expand All @@ -92,6 +93,7 @@ jobs:
pre-script: packaging/pre_build_script.sh
post-script: packaging/post_build_script.sh
smoke-test-script: packaging/smoke_test_script.sh
display-name: L0 dynamo converter tests
uses: ./.github/workflows/linux-test.yml
with:
job-name: L0-dynamo-converter-tests
Expand All @@ -109,7 +111,7 @@ jobs:
popd

L0-dynamo-core-tests:
name: L0 dynamo core tests
name: ${{ matrix.display-name }}
needs: [filter-matrix, build]
if: ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
strategy:
Expand All @@ -121,6 +123,7 @@ jobs:
pre-script: packaging/pre_build_script.sh
post-script: packaging/post_build_script.sh
smoke-test-script: packaging/smoke_test_script.sh
display-name: L0 dynamo core tests
uses: ./.github/workflows/linux-test.yml
with:
job-name: L0-dynamo-core-tests
Expand All @@ -136,12 +139,12 @@ jobs:
cd tests/py
cd dynamo
python -m pytest -ra -n 8 --junitxml=${RUNNER_TEST_RESULTS_DIR}/l0_dynamo_core_runtime_tests_results.xml runtime/test_000_*
python -m pytest -ra -n 8 --junitxml=${RUNNER_TEST_RESULTS_DIR}/l0_dynamo_core_partitioning_tests_results.xml partitioning/test_000_*
python -m pytest -ra -n 1 --junitxml=${RUNNER_TEST_RESULTS_DIR}/l0_dynamo_core_partitioning_tests_results.xml partitioning/test_000_*
python -m pytest -ra -n 8 --junitxml=${RUNNER_TEST_RESULTS_DIR}/l0_dynamo_core_lowering_tests_results.xml lowering/
popd

L0-py-core-tests:
name: L0 core python tests
name: ${{ matrix.display-name }}
needs: [filter-matrix, build]
if: ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
strategy:
Expand All @@ -153,6 +156,7 @@ jobs:
pre-script: packaging/pre_build_script.sh
post-script: packaging/post_build_script.sh
smoke-test-script: packaging/smoke_test_script.sh
display-name: L0 core python tests
uses: ./.github/workflows/linux-test.yml
with:
job-name: L0-py-core-tests
Expand All @@ -170,7 +174,7 @@ jobs:
popd

L0-torchscript-tests:
name: L0 torchscript tests
name: ${{ matrix.display-name }}
needs: [filter-matrix, build]
if: ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
strategy:
Expand All @@ -182,6 +186,7 @@ jobs:
pre-script: packaging/pre_build_script.sh
post-script: packaging/post_build_script.sh
smoke-test-script: packaging/smoke_test_script.sh
display-name: L0 torchscript tests
uses: ./.github/workflows/linux-test.yml
with:
job-name: L0-torchscript-tests
Expand All @@ -203,7 +208,7 @@ jobs:
popd

L1-dynamo-core-tests:
name: L1 dynamo core tests
name: ${{ matrix.display-name }}
needs: [filter-matrix, build, L0-dynamo-converter-tests, L0-dynamo-core-tests, L0-py-core-tests, L0-torchscript-tests]
if: ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
strategy:
Expand All @@ -215,6 +220,7 @@ jobs:
pre-script: packaging/pre_build_script.sh
post-script: packaging/post_build_script.sh
smoke-test-script: packaging/smoke_test_script.sh
display-name: L1 dynamo core tests
uses: ./.github/workflows/linux-test.yml
with:
job-name: L1-dynamo-core-tests
Expand All @@ -234,7 +240,7 @@ jobs:
popd

L1-dynamo-compile-tests:
name: L1 dynamo compile tests
name: ${{ matrix.display-name }}
needs: [filter-matrix, build, L0-dynamo-converter-tests, L0-dynamo-core-tests, L0-py-core-tests, L0-torchscript-tests]
if: ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
strategy:
Expand All @@ -246,6 +252,7 @@ jobs:
pre-script: packaging/pre_build_script.sh
post-script: packaging/post_build_script.sh
smoke-test-script: packaging/smoke_test_script.sh
display-name: L1 dynamo compile tests
uses: ./.github/workflows/linux-test.yml
with:
job-name: L1-dynamo-compile-tests
Expand All @@ -263,7 +270,7 @@ jobs:
popd

L1-torch-compile-tests:
name: L1 torch compile tests
name: ${{ matrix.display-name }}
needs: [filter-matrix, build, L0-dynamo-converter-tests, L0-dynamo-core-tests, L0-py-core-tests, L0-torchscript-tests]
if: ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
strategy:
Expand All @@ -275,6 +282,7 @@ jobs:
pre-script: packaging/pre_build_script.sh
post-script: packaging/post_build_script.sh
smoke-test-script: packaging/smoke_test_script.sh
display-name: L1 torch compile tests
uses: ./.github/workflows/linux-test.yml
with:
job-name: L1-torch-compile-tests
Expand All @@ -294,7 +302,7 @@ jobs:
popd

L1-torchscript-tests:
name: L1 torch script tests
name: ${{ matrix.display-name }}
needs: [filter-matrix, build, L0-dynamo-core-tests, L0-dynamo-converter-tests, L0-py-core-tests, L0-torchscript-tests]
if: ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
strategy:
Expand All @@ -306,6 +314,7 @@ jobs:
pre-script: packaging/pre_build_script.sh
post-script: packaging/post_build_script.sh
smoke-test-script: packaging/smoke_test_script.sh
display-name: L1 torch script tests
uses: ./.github/workflows/linux-test.yml
with:
job-name: L1-torchscript-tests
Expand All @@ -327,7 +336,7 @@ jobs:
popd

L2-torch-compile-tests:
name: L2 torch compile tests
name: ${{ matrix.display-name }}
needs: [filter-matrix, build, L1-torch-compile-tests, L1-dynamo-compile-tests, L1-dynamo-core-tests, L1-torchscript-tests]
if: ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
strategy:
Expand All @@ -339,6 +348,7 @@ jobs:
pre-script: packaging/pre_build_script.sh
post-script: packaging/post_build_script.sh
smoke-test-script: packaging/smoke_test_script.sh
display-name: L2 torch compile tests
uses: ./.github/workflows/linux-test.yml
with:
job-name: L2-torch-compile-tests
Expand All @@ -357,7 +367,7 @@ jobs:
popd

L2-dynamo-compile-tests:
name: L2 dynamo compile tests
name: ${{ matrix.display-name }}
needs: [filter-matrix, build, L1-dynamo-compile-tests, L1-dynamo-core-tests, L1-torch-compile-tests, L1-torchscript-tests]
if: ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
strategy:
Expand All @@ -369,6 +379,7 @@ jobs:
pre-script: packaging/pre_build_script.sh
post-script: packaging/post_build_script.sh
smoke-test-script: packaging/smoke_test_script.sh
display-name: L2 dynamo compile tests
uses: ./.github/workflows/linux-test.yml
with:
job-name: L2-dynamo-compile-tests
Expand All @@ -387,7 +398,7 @@ jobs:
popd

L2-dynamo-core-tests:
name: L2 dynamo core tests
name: ${{ matrix.display-name }}
needs: [filter-matrix, build, L1-dynamo-core-tests, L1-dynamo-compile-tests, L1-torch-compile-tests, L1-torchscript-tests]
if: ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
strategy:
Expand All @@ -399,6 +410,7 @@ jobs:
pre-script: packaging/pre_build_script.sh
post-script: packaging/post_build_script.sh
smoke-test-script: packaging/smoke_test_script.sh
display-name: L2 dynamo core tests
uses: ./.github/workflows/linux-test.yml
with:
job-name: L2-dynamo-core-tests
Expand All @@ -416,7 +428,7 @@ jobs:
popd

L2-dynamo-plugin-tests:
name: L2 dynamo plugin tests
name: ${{ matrix.display-name }}
needs: [filter-matrix, build, L1-dynamo-core-tests, L1-dynamo-compile-tests, L1-torch-compile-tests, L1-torchscript-tests]
if: ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
strategy:
Expand All @@ -428,6 +440,7 @@ jobs:
pre-script: packaging/pre_build_script.sh
post-script: packaging/post_build_script.sh
smoke-test-script: packaging/smoke_test_script.sh
display-name: L2 dynamo plugin tests
uses: ./.github/workflows/linux-test.yml
with:
job-name: L2-dynamo-plugin-tests
Expand All @@ -448,7 +461,7 @@ jobs:
popd

L2-torchscript-tests:
name: L2 torch script tests
name: ${{ matrix.display-name }}
needs: [filter-matrix, build, L1-dynamo-core-tests, L1-dynamo-compile-tests, L1-torch-compile-tests, L1-torchscript-tests]
if: ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
strategy:
Expand All @@ -460,6 +473,7 @@ jobs:
pre-script: packaging/pre_build_script.sh
post-script: packaging/post_build_script.sh
smoke-test-script: packaging/smoke_test_script.sh
display-name: L2 torch script tests
uses: ./.github/workflows/linux-test.yml
with:
job-name: L2-torchscript-tests
Expand All @@ -481,7 +495,7 @@ jobs:
popd

L2-dynamo-distributed-tests:
name: L2 dynamo distributed tests
name: ${{ matrix.display-name }}
needs: [filter-matrix, build, L1-dynamo-core-tests, L1-dynamo-compile-tests, L1-torch-compile-tests, L1-torchscript-tests]
if: ${{ (github.ref_name == 'main' || github.ref_name == 'nightly' || contains(github.event.pull_request.labels.*.name, 'Force All Tests[L0+L1+L2]')) && always() || success() }}
strategy:
Expand All @@ -493,6 +507,7 @@ jobs:
pre-script: packaging/pre_build_script.sh
post-script: packaging/post_build_script.sh
smoke-test-script: packaging/smoke_test_script.sh
display-name: L2 dynamo distributed tests
uses: ./.github/workflows/linux-test.yml
with:
job-name: L2-dynamo-distributed-tests
Expand Down
Loading
Loading