Skip to content

Commit 646a515

Browse files
lanluo-nvidialanluo-nvidia
committed
Merge branch 'main' into lluo/tensorrt_test_workflow
2 parents e9af038 + b4de166 commit 646a515

File tree

200 files changed

+3629
-877
lines changed

Some content is hidden

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

200 files changed

+3629
-877
lines changed

‎.github/scripts/filter-matrix.py

Lines changed: 35 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,35 @@
1+
#!/usr/bin/env python3
2+
3+
import argparse
4+
import json
5+
import sys
6+
7+
disabled_python_versions = "3.13"
8+
9+
10+
def main(args: list[str]) -> None:
11+
parser = argparse.ArgumentParser()
12+
parser.add_argument(
13+
"--matrix",
14+
help="matrix blob",
15+
type=str,
16+
default="",
17+
)
18+
19+
options = parser.parse_args(args)
20+
21+
if options.matrix == "":
22+
raise Exception("--matrix needs to be provided")
23+
matrix_dict = json.loads(options.matrix)
24+
includes = matrix_dict["include"]
25+
filtered_includes = []
26+
for item in includes:
27+
if item["python_version"] not in disabled_python_versions:
28+
filtered_includes.append(item)
29+
filtered_matrix_dict = {}
30+
filtered_matrix_dict["include"] = filtered_includes
31+
print(json.dumps(filtered_matrix_dict))
32+
33+
34+
if __name__ == "__main__":
35+
main(sys.argv[1:])

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

Lines changed: 39 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,29 @@ jobs:
2424
with-rocm: false
2525
with-cpu: false
2626

27+
filter-matrix:
28+
needs: [generate-matrix]
29+
outputs:
30+
matrix: ${{ steps.generate.outputs.matrix }}
31+
runs-on: ubuntu-latest
32+
steps:
33+
- uses: actions/setup-python@v4
34+
with:
35+
python-version: '3.10'
36+
- uses: actions/checkout@v3
37+
with:
38+
repository: pytorch/tensorrt
39+
- name: Generate release matrix
40+
id: generate
41+
run: |
42+
set -eou pipefail
43+
MATRIX_BLOB=${{ toJSON(needs.generate-matrix.outputs.matrix) }}
44+
MATRIX_BLOB="$(python3 .github/scripts/filter-matrix.py --matrix "${MATRIX_BLOB}")"
45+
echo "${MATRIX_BLOB}"
46+
echo "matrix=${MATRIX_BLOB}" >> "${GITHUB_OUTPUT}"
47+
2748
build:
28-
needs: generate-matrix
49+
needs: filter-matrix
2950
permissions:
3051
id-token: write
3152
contents: read
@@ -46,7 +67,7 @@ jobs:
4667
ref: ""
4768
test-infra-repository: pytorch/test-infra
4869
test-infra-ref: main
49-
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
70+
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
5071
pre-script: ${{ matrix.pre-script }}
5172
env-var-script: ${{ matrix.env-var-script }}
5273
post-script: ${{ matrix.post-script }}
@@ -56,7 +77,7 @@ jobs:
5677

5778
tests-py-torchscript-fe:
5879
name: Test torchscript frontend [Python]
59-
needs: [generate-matrix, build]
80+
needs: [filter-matrix, build]
6081
strategy:
6182
fail-fast: false
6283
matrix:
@@ -73,7 +94,7 @@ jobs:
7394
ref: ""
7495
test-infra-repository: pytorch/test-infra
7596
test-infra-ref: main
76-
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
97+
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
7798
pre-script: ${{ matrix.pre-script }}
7899
script: |
79100
export USE_HOST_DEPS=1
@@ -92,7 +113,7 @@ jobs:
92113
93114
tests-py-dynamo-converters:
94115
name: Test dynamo converters [Python]
95-
needs: [generate-matrix, build]
116+
needs: [filter-matrix, build]
96117
strategy:
97118
fail-fast: false
98119
matrix:
@@ -109,7 +130,7 @@ jobs:
109130
ref: ""
110131
test-infra-repository: pytorch/test-infra
111132
test-infra-ref: main
112-
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
133+
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
113134
pre-script: ${{ matrix.pre-script }}
114135
script: |
115136
export USE_HOST_DEPS=1
@@ -121,7 +142,7 @@ jobs:
121142
122143
tests-py-dynamo-fe:
123144
name: Test dynamo frontend [Python]
124-
needs: [generate-matrix, build]
145+
needs: [filter-matrix, build]
125146
strategy:
126147
fail-fast: false
127148
matrix:
@@ -138,7 +159,7 @@ jobs:
138159
ref: ""
139160
test-infra-repository: pytorch/test-infra
140161
test-infra-ref: main
141-
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
162+
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
142163
pre-script: ${{ matrix.pre-script }}
143164
script: |
144165
export USE_HOST_DEPS=1
@@ -150,7 +171,7 @@ jobs:
150171
151172
tests-py-dynamo-serde:
152173
name: Test dynamo export serde [Python]
153-
needs: [generate-matrix, build]
174+
needs: [filter-matrix, build]
154175
strategy:
155176
fail-fast: false
156177
matrix:
@@ -167,7 +188,7 @@ jobs:
167188
ref: ""
168189
test-infra-repository: pytorch/test-infra
169190
test-infra-ref: main
170-
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
191+
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
171192
pre-script: ${{ matrix.pre-script }}
172193
script: |
173194
export USE_HOST_DEPS=1
@@ -179,7 +200,7 @@ jobs:
179200
180201
tests-py-torch-compile-be:
181202
name: Test torch compile backend [Python]
182-
needs: [generate-matrix, build]
203+
needs: [filter-matrix, build]
183204
strategy:
184205
fail-fast: false
185206
matrix:
@@ -196,7 +217,7 @@ jobs:
196217
ref: ""
197218
test-infra-repository: pytorch/test-infra
198219
test-infra-ref: main
199-
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
220+
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
200221
pre-script: ${{ matrix.pre-script }}
201222
script: |
202223
export USE_HOST_DEPS=1
@@ -210,7 +231,7 @@ jobs:
210231
211232
tests-py-dynamo-core:
212233
name: Test dynamo core [Python]
213-
needs: [generate-matrix, build]
234+
needs: [filter-matrix, build]
214235
strategy:
215236
fail-fast: false
216237
matrix:
@@ -227,7 +248,7 @@ jobs:
227248
ref: ""
228249
test-infra-repository: pytorch/test-infra
229250
test-infra-ref: main
230-
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
251+
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
231252
pre-script: ${{ matrix.pre-script }}
232253
script: |
233254
export USE_HOST_DEPS=1
@@ -241,7 +262,7 @@ jobs:
241262
242263
tests-py-dynamo-cudagraphs:
243264
name: Test dynamo cudagraphs [Python]
244-
needs: [generate-matrix, build]
265+
needs: [filter-matrix, build]
245266
strategy:
246267
fail-fast: false
247268
matrix:
@@ -258,7 +279,7 @@ jobs:
258279
ref: ""
259280
test-infra-repository: pytorch/test-infra
260281
test-infra-ref: main
261-
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
282+
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
262283
pre-script: ${{ matrix.pre-script }}
263284
script: |
264285
export USE_HOST_DEPS=1
@@ -272,7 +293,7 @@ jobs:
272293
273294
tests-py-core:
274295
name: Test core [Python]
275-
needs: [generate-matrix, build]
296+
needs: [filter-matrix, build]
276297
strategy:
277298
fail-fast: false
278299
matrix:
@@ -289,7 +310,7 @@ jobs:
289310
ref: ""
290311
test-infra-repository: pytorch/test-infra
291312
test-infra-ref: main
292-
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
313+
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
293314
pre-script: ${{ matrix.pre-script }}
294315
script: |
295316
export USE_HOST_DEPS=1

‎docs/_cpp_api/classtorch__tensorrt_1_1DataType.html

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
<meta name="viewport" content="width=device-width, initial-scale=1.0">
1212

13-
<title>Class DataType &mdash; Torch-TensorRT v2.6.0.dev0+92bf700 documentation</title>
13+
<title>Class DataType &mdash; Torch-TensorRT v2.6.0.dev0+233d0bf documentation</title>
1414

1515

1616

@@ -275,7 +275,7 @@
275275

276276

277277
<div class="version">
278-
v2.6.0.dev0+92bf700
278+
v2.6.0.dev0+233d0bf
279279
</div>
280280

281281

‎docs/_cpp_api/classtorch__tensorrt_1_1Device_1_1DeviceType.html

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
<meta name="viewport" content="width=device-width, initial-scale=1.0">
1212

13-
<title>Class Device::DeviceType &mdash; Torch-TensorRT v2.6.0.dev0+92bf700 documentation</title>
13+
<title>Class Device::DeviceType &mdash; Torch-TensorRT v2.6.0.dev0+233d0bf documentation</title>
1414

1515

1616

@@ -275,7 +275,7 @@
275275

276276

277277
<div class="version">
278-
v2.6.0.dev0+92bf700
278+
v2.6.0.dev0+233d0bf
279279
</div>
280280

281281

‎docs/_cpp_api/classtorch__tensorrt_1_1TensorFormat.html

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
<meta name="viewport" content="width=device-width, initial-scale=1.0">
1212

13-
<title>Class TensorFormat &mdash; Torch-TensorRT v2.6.0.dev0+92bf700 documentation</title>
13+
<title>Class TensorFormat &mdash; Torch-TensorRT v2.6.0.dev0+233d0bf documentation</title>
1414

1515

1616

@@ -275,7 +275,7 @@
275275

276276

277277
<div class="version">
278-
v2.6.0.dev0+92bf700
278+
v2.6.0.dev0+233d0bf
279279
</div>
280280

281281

‎docs/_cpp_api/classtorch__tensorrt_1_1ptq_1_1Int8CacheCalibrator.html

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
<meta name="viewport" content="width=device-width, initial-scale=1.0">
1212

13-
<title>Template Class Int8CacheCalibrator &mdash; Torch-TensorRT v2.6.0.dev0+92bf700 documentation</title>
13+
<title>Template Class Int8CacheCalibrator &mdash; Torch-TensorRT v2.6.0.dev0+233d0bf documentation</title>
1414

1515

1616

@@ -275,7 +275,7 @@
275275

276276

277277
<div class="version">
278-
v2.6.0.dev0+92bf700
278+
v2.6.0.dev0+233d0bf
279279
</div>
280280

281281

‎docs/_cpp_api/classtorch__tensorrt_1_1ptq_1_1Int8Calibrator.html

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
<meta name="viewport" content="width=device-width, initial-scale=1.0">
1212

13-
<title>Template Class Int8Calibrator &mdash; Torch-TensorRT v2.6.0.dev0+92bf700 documentation</title>
13+
<title>Template Class Int8Calibrator &mdash; Torch-TensorRT v2.6.0.dev0+233d0bf documentation</title>
1414

1515

1616

@@ -275,7 +275,7 @@
275275

276276

277277
<div class="version">
278-
v2.6.0.dev0+92bf700
278+
v2.6.0.dev0+233d0bf
279279
</div>
280280

281281

‎docs/_cpp_api/define_macros_8h_1a18d295a837ac71add5578860b55e5502.html

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
<meta name="viewport" content="width=device-width, initial-scale=1.0">
1212

13-
<title>Define STR &mdash; Torch-TensorRT v2.6.0.dev0+92bf700 documentation</title>
13+
<title>Define STR &mdash; Torch-TensorRT v2.6.0.dev0+233d0bf documentation</title>
1414

1515

1616

@@ -275,7 +275,7 @@
275275

276276

277277
<div class="version">
278-
v2.6.0.dev0+92bf700
278+
v2.6.0.dev0+233d0bf
279279
</div>
280280

281281

‎docs/_cpp_api/define_macros_8h_1a282fd3c0b1c3a215148ae372070e1268.html

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
<meta name="viewport" content="width=device-width, initial-scale=1.0">
1212

13-
<title>Define TORCH_TENSORRT_PATCH_VERSION &mdash; Torch-TensorRT v2.6.0.dev0+92bf700 documentation</title>
13+
<title>Define TORCH_TENSORRT_PATCH_VERSION &mdash; Torch-TensorRT v2.6.0.dev0+233d0bf documentation</title>
1414

1515

1616

@@ -275,7 +275,7 @@
275275

276276

277277
<div class="version">
278-
v2.6.0.dev0+92bf700
278+
v2.6.0.dev0+233d0bf
279279
</div>
280280

281281

‎docs/_cpp_api/define_macros_8h_1a31398a6d4d27e28817afb0f0139e909e.html

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
<meta name="viewport" content="width=device-width, initial-scale=1.0">
1212

13-
<title>Define TORCH_TENSORRT_MAJOR_VERSION &mdash; Torch-TensorRT v2.6.0.dev0+92bf700 documentation</title>
13+
<title>Define TORCH_TENSORRT_MAJOR_VERSION &mdash; Torch-TensorRT v2.6.0.dev0+233d0bf documentation</title>
1414

1515

1616

@@ -275,7 +275,7 @@
275275

276276

277277
<div class="version">
278-
v2.6.0.dev0+92bf700
278+
v2.6.0.dev0+233d0bf
279279
</div>
280280

281281

0 commit comments

Comments
 (0)