Skip to content

Commit 479928f

Browse files
[mlir-tensorrt][cmake] Add TRT 10.12 download support
1 parent b854eef commit 479928f

File tree

1 file changed

+11
-5
lines changed

1 file changed

+11
-5
lines changed

mlir-tensorrt/build_tools/cmake/TensorRTDownloadURL.cmake

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,10 @@ function(mtrt_get_tensorrt_download_url ARG_VERSION OS_NAME TARGET_ARCH ARG_OUT_
7676
set(ARG_VERSION "10.9.0.34")
7777
endif()
7878

79+
if(ARG_VERSION VERSION_EQUAL "10.12")
80+
set(ARG_VERSION "10.12.0.36")
81+
endif()
82+
7983
set(downloadable_versions
8084
"8.6.1.6"
8185
"9.0.1.4" "9.1.0.4" "9.2.0.5"
@@ -92,6 +96,7 @@ function(mtrt_get_tensorrt_download_url ARG_VERSION OS_NAME TARGET_ARCH ARG_OUT_
9296
"10.7.0.23"
9397
"10.8.0.43"
9498
"10.9.0.34"
99+
"10.12.0.36"
95100
)
96101

97102
if(NOT ARG_VERSION IN_LIST downloadable_versions)
@@ -146,18 +151,19 @@ function(mtrt_get_tensorrt_download_url ARG_VERSION OS_NAME TARGET_ARCH ARG_OUT_
146151
endif()
147152
endif()
148153

149-
# Handle TensorRT 8 versions. These are publicly accessible download links.
150-
if(ARG_VERSION VERSION_LESS 9.0)
151-
set(TRT_CUDA_VERSION 12.0)
152-
elseif(ARG_VERSION VERSION_GREATER 10.2 AND
154+
if(ARG_VERSION VERSION_GREATER 10.2 AND
153155
ARG_VERSION VERSION_LESS 10.4)
154156
set(TRT_CUDA_VERSION 12.5)
155157
elseif(
156158
ARG_VERSION VERSION_GREATER 10.4 AND
157159
ARG_VERSION VERSION_LESS 10.8)
158160
set(TRT_CUDA_VERSION 12.6)
159-
elseif(ARG_VERSION VERSION_GREATER 10.7)
161+
elseif(ARG_VERSION VERSION_GREATER 10.7
162+
AND ARG_VERSION VERSION_LESS 10.10)
160163
set(TRT_CUDA_VERSION 12.8)
164+
elseif(ARG_VERSION VERSION_GREATER 10.10
165+
AND ARG_VERSION VERSION_LESS 10.13)
166+
set(TRT_CUDA_VERSION 12.9)
161167
endif()
162168

163169
# Handle TRT 8 versions.

0 commit comments

Comments
 (0)