@@ -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