Skip to content

Commit ba4fa05

Browse files
author
pytorchbot
committed
2026-01-05 nightly release (ec2b7ea)
1 parent b4d4d16 commit ba4fa05

File tree

191 files changed

+2972
-2544
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

191 files changed

+2972
-2544
lines changed

.github/pr-labels.yml

Lines changed: 58 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -1,57 +1,89 @@
11
"component: torchtrtc":
2-
- cpp/torchtrtc/**/*
2+
- changed-files:
3+
- any-glob-to-any-file:
4+
- cpp/torchtrtc/**/*
35

46
"component: api [C++]":
5-
- cpp/**/*
7+
- changed-files:
8+
- any-glob-to-any-file:
9+
- cpp/**/*
610

711
"component: api [Python]":
8-
- py/**/*
12+
- changed-files:
13+
- any-glob-to-any-file:
14+
- py/**/*
915

1016
"component: core":
11-
- core/**/*
12-
- py/torch_tensorrt/dynamo/**/*
17+
- changed-files:
18+
- any-glob-to-any-file:
19+
- core/**/*
20+
- py/torch_tensorrt/dynamo/**/*
1321

1422
"component: conversion":
15-
- core/conversion/**/*
16-
- py/torch_tensorrt/dynamo/conversion/**/*
23+
- changed-files:
24+
- any-glob-to-any-file:
25+
- core/conversion/**/*
26+
- py/torch_tensorrt/dynamo/conversion/**/*
1727

1828
"component: converters":
19-
- core/conversion/converters/**/*
20-
- py/torch_tensorrt/dynamo/conversion/impl/**/*
29+
- changed-files:
30+
- any-glob-to-any-file:
31+
- core/conversion/converters/**/*
32+
- py/torch_tensorrt/dynamo/conversion/impl/**/*
2133

2234
"component: evaluators":
23-
- core/conversion/evaluators/**/*
35+
- changed-files:
36+
- any-glob-to-any-file:
37+
- core/conversion/evaluators/**/*
2438

2539
"component: fx":
26-
- py/torch_tensorrt/fx/**/*
40+
- changed-files:
41+
- any-glob-to-any-file:
42+
- py/torch_tensorrt/fx/**/*
2743

2844
"component: dynamo":
29-
- py/torch_tensorrt/dynamo/**/*
45+
- changed-files:
46+
- any-glob-to-any-file:
47+
- py/torch_tensorrt/dynamo/**/*
3048

3149
"component: torch_compile":
32-
- py/torch_tensorrt/dynamo/backend/*
50+
- changed-files:
51+
- any-glob-to-any-file:
52+
- py/torch_tensorrt/dynamo/backend/*
3353

3454
"component: partitioning":
35-
- core/partitioning/**/*
55+
- changed-files:
56+
- any-glob-to-any-file:
57+
- core/partitioning/**/*
3658

3759
"component: runtime":
38-
- core/runtime/**/*
39-
- py/torch_tensorrt/dynamo/runtime/**/*
60+
- changed-files:
61+
- any-glob-to-any-file:
62+
- core/runtime/**/*
63+
- py/torch_tensorrt/dynamo/runtime/**/*
4064

4165
"component: lowering":
42-
- core/lowering/**/*
43-
- py/torch_tensorrt/dynamo/lowering/**/*
66+
- changed-files:
67+
- any-glob-to-any-file:
68+
- core/lowering/**/*
69+
- py/torch_tensorrt/dynamo/lowering/**/*
4470

4571
"component: tests":
46-
- tests/**/*
72+
- changed-files:
73+
- any-glob-to-any-file:
74+
- tests/**/*
4775

4876
"component: build system":
49-
- MODULE.bazel
50-
- BUILD
51-
- pyproject.toml
52-
- setup.py
53-
- toolchain/**/*
77+
- changed-files:
78+
- any-glob-to-any-file:
79+
- MODULE.bazel
80+
- BUILD
81+
- pyproject.toml
82+
- setup.py
83+
- toolchain/**/*
5484

5585
"documentation":
56-
- docs/**/*
57-
- docsrc/**/*
86+
- changed-files:
87+
- any-glob-to-any-file:
88+
- docs/**/*
89+
- docsrc/**/*

.github/scripts/filter-matrix.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,10 @@
1313
# jetpack 6.2 only officially supports python 3.10 and cu126
1414
jetpack_python_versions: List[str] = ["3.10"]
1515
jetpack_cuda_versions: List[str] = ["cu126"]
16+
# rtx 1.2 currently only supports cu129 and cu130
17+
rtx_cuda_versions: List[str] = ["cu129", "cu130"]
18+
# trt 10.14.1 currently only supports cu129 and cu130
19+
trt_cuda_versions: List[str] = ["cu129", "cu130"]
1620

1721
jetpack_container_image: str = "nvcr.io/nvidia/l4t-jetpack:r36.4.0"
1822
sbsa_container_image: str = "quay.io/pypa/manylinux_2_39_aarch64"
@@ -32,6 +36,7 @@ def filter_matrix_item(
3236
item: Dict[str, Any],
3337
is_jetpack: bool,
3438
limit_pr_builds: bool,
39+
use_rtx: bool,
3540
) -> bool:
3641
"""Filter a single matrix item based on the build type and requirements."""
3742
if item["python_version"] in disabled_python_versions:
@@ -51,6 +56,12 @@ def filter_matrix_item(
5156
return True
5257
return False
5358
else:
59+
if use_rtx:
60+
if item["desired_cuda"] not in rtx_cuda_versions:
61+
return False
62+
else:
63+
if item["desired_cuda"] not in trt_cuda_versions:
64+
return False
5465
if item["gpu_arch_type"] == "cuda-aarch64":
5566
# pytorch image:pytorch/manylinuxaarch64-builder:cuda12.8 comes with glibc2.28
5667
# however, TensorRT requires glibc2.31 on aarch64 platform
@@ -85,6 +96,14 @@ def main(args: list[str]) -> None:
8596
default=os.getenv("LIMIT_PR_BUILDS", "false"),
8697
)
8798

99+
parser.add_argument(
100+
"--use-rtx",
101+
help="use rtx",
102+
type=str,
103+
choices=["true", "false"],
104+
default="false",
105+
)
106+
88107
options = parser.parse_args(args)
89108
if options.matrix == "":
90109
raise ValueError("--matrix needs to be provided")
@@ -105,6 +124,7 @@ def main(args: list[str]) -> None:
105124
item,
106125
options.jetpack == "true",
107126
options.limit_pr_builds == "true",
127+
options.use_rtx == "true",
108128
):
109129
filtered_includes.append(item)
110130

.github/workflows/assigner.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ jobs:
2222
runs-on: ubuntu-latest
2323
steps:
2424
- name: Checkout
25-
uses: actions/checkout@v4
25+
uses: actions/checkout@v6
2626

2727
- name: Assign
2828
uses: ./.github/actions/assigner

.github/workflows/blossom-ci.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,15 +50,15 @@ jobs:
5050
runs-on: ubuntu-latest
5151
steps:
5252
- name: Checkout code
53-
uses: actions/checkout@v2
53+
uses: actions/checkout@v6
5454
with:
5555
repository: ${{ fromJson(needs.Authorization.outputs.args).repo }}
5656
ref: ${{ fromJson(needs.Authorization.outputs.args).ref }}
5757
lfs: 'true'
5858

5959
# repo specific steps
6060
#- name: Setup java
61-
# uses: actions/setup-java@v1
61+
# uses: actions/setup-java@v5
6262
# with:
6363
# java-version: 1.8
6464

.github/workflows/build-tensorrt-linux.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -114,13 +114,13 @@ jobs:
114114
rm -rf "${RUNNER_TEMP}/*"
115115
fi
116116
echo "::endgroup::"
117-
- uses: actions/checkout@v4
117+
- uses: actions/checkout@v6
118118
with:
119119
# Support the use case where we need to checkout someone's fork
120120
repository: ${{ inputs.test-infra-repository }}
121121
ref: ${{ inputs.test-infra-ref }}
122122
path: test-infra
123-
- uses: actions/checkout@v4
123+
- uses: actions/checkout@v6
124124
if: ${{ env.ARCH == 'aarch64' }}
125125
with:
126126
# Support the use case where we need to checkout someone's fork
@@ -212,7 +212,7 @@ jobs:
212212
# NB: Only upload to GitHub after passing smoke tests
213213
- name: Upload wheel to GitHub
214214
continue-on-error: true
215-
uses: actions/upload-artifact@v4
215+
uses: actions/upload-artifact@v6
216216
with:
217217
name: ${{ env.UPLOAD_ARTIFACT_NAME }}
218218
path: ${{ inputs.repository }}/dist

.github/workflows/build-tensorrt-windows.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@ jobs:
100100
# to have a conversation
101101
timeout-minutes: 120
102102
steps:
103-
- uses: actions/checkout@v4
103+
- uses: actions/checkout@v6
104104
with:
105105
# Support the use case where we need to checkout someone's fork
106106
repository: ${{ inputs.test-infra-repository }}
@@ -216,7 +216,7 @@ jobs:
216216
# NB: Only upload to GitHub after passing smoke tests
217217
- name: Upload wheel to GitHub
218218
continue-on-error: true
219-
uses: actions/upload-artifact@v4
219+
uses: actions/upload-artifact@v6
220220
with:
221221
name: ${{ env.UPLOAD_ARTIFACT_NAME }}
222222
path: ${{ inputs.repository }}/dist/

.github/workflows/build-test-linux-aarch64-jetpack.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,10 +30,10 @@ jobs:
3030
matrix: ${{ steps.filter.outputs.matrix }}
3131
runs-on: ubuntu-latest
3232
steps:
33-
- uses: actions/setup-python@v5
33+
- uses: actions/setup-python@v6
3434
with:
3535
python-version: "3.11"
36-
- uses: actions/checkout@v4
36+
- uses: actions/checkout@v6
3737
with:
3838
repository: pytorch/tensorrt
3939
- name: Filter matrix

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,10 +30,10 @@ jobs:
3030
matrix: ${{ steps.filter.outputs.matrix }}
3131
runs-on: ubuntu-latest
3232
steps:
33-
- uses: actions/setup-python@v5
33+
- uses: actions/setup-python@v6
3434
with:
3535
python-version: "3.11"
36-
- uses: actions/checkout@v4
36+
- uses: actions/checkout@v6
3737
with:
3838
repository: pytorch/tensorrt
3939
- name: Filter matrix

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,10 +31,10 @@ jobs:
3131
matrix: ${{ steps.generate.outputs.matrix }}
3232
runs-on: ubuntu-latest
3333
steps:
34-
- uses: actions/setup-python@v5
34+
- uses: actions/setup-python@v6
3535
with:
3636
python-version: '3.11'
37-
- uses: actions/checkout@v4
37+
- uses: actions/checkout@v6
3838
with:
3939
repository: pytorch/tensorrt
4040
- name: Generate matrix

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,18 +30,18 @@ jobs:
3030
matrix: ${{ steps.generate.outputs.matrix }}
3131
runs-on: ubuntu-latest
3232
steps:
33-
- uses: actions/setup-python@v5
33+
- uses: actions/setup-python@v6
3434
with:
3535
python-version: '3.11'
36-
- uses: actions/checkout@v4
36+
- uses: actions/checkout@v6
3737
with:
3838
repository: pytorch/tensorrt
3939
- name: Generate matrix
4040
id: generate
4141
run: |
4242
set -eou pipefail
4343
MATRIX_BLOB=${{ toJSON(needs.generate-matrix.outputs.matrix) }}
44-
MATRIX_BLOB="$(python3 .github/scripts/filter-matrix.py --matrix "${MATRIX_BLOB}")"
44+
MATRIX_BLOB="$(python3 .github/scripts/filter-matrix.py --use-rtx true --matrix "${MATRIX_BLOB}")"
4545
echo "${MATRIX_BLOB}"
4646
echo "matrix=${MATRIX_BLOB}" >> "${GITHUB_OUTPUT}"
4747

0 commit comments

Comments
 (0)