Skip to content

Commit 95932e8

Browse files
committed
add more tensorrt version
1 parent bfedc3c commit 95932e8

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

.github/scripts/generate-tensorrt-test-matrix.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,14 @@
4444
"urls": "https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.9.0/zip/TensorRT-10.9.0.34.Windows.win10.cuda-12.8.zip",
4545
"strip_prefix": "TensorRT-10.9.0.34",
4646
},
47+
"10.10.0": {
48+
"urls": "https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.10.0/zip/TensorRT-10.10.0.31.Windows.win10.cuda-12.9.zip",
49+
"strip_prefix": "TensorRT-10.10.0.31",
50+
},
51+
"10.11.0": {
52+
"urls": "https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.11.0/zip/TensorRT-10.11.0.33.Windows.win10.cuda-12.9.zip",
53+
"strip_prefix": "TensorRT-10.11.0.33",
54+
},
4755
},
4856
"linux": {
4957
"10.3.0": {
@@ -62,13 +70,23 @@
6270
"urls": "https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.9.0/tars/TensorRT-10.9.0.34.Linux.x86_64-gnu.cuda-12.8.tar.gz",
6371
"strip_prefix": "TensorRT-10.9.0.34",
6472
},
73+
"10.10.0": {
74+
"urls": "https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.10.0/tars/TensorRT-10.10.0.31.Linux.x86_64-gnu.cuda-12.9.tar.gz",
75+
"strip_prefix": "TensorRT-10.10.0.31",
76+
},
77+
"10.11.0": {
78+
"urls": "https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.11.0/tars/TensorRT-10.11.0.33.Linux.x86_64-gnu.cuda-12.9.tar.gz",
79+
"strip_prefix": "TensorRT-10.11.0.33",
80+
},
6581
},
6682
}
6783

6884

6985
def check_new_tensorrt_version(
7086
major: int = 10, patch_from: int = 0
7187
) -> tuple[bool, str, str, str, str]:
88+
breakpoint()
89+
7290
def check_file_availability(url: str) -> bool:
7391
try:
7492
response = requests.head(url, allow_redirects=True)

0 commit comments

Comments
 (0)