|
36 | 36 |
|
37 | 37 |
|
38 | 38 | file(READ ${TENSORRT_INCLUDE_DIR}/NvInferVersion.h NVINFER_VER_CONTENT)
|
39 |
| - string(REGEX MATCH "define NV_TENSORRT_MAJOR * +([0-9]+)" NV_TENSORRT_MAJOR "${NVINFER_VER_CONTENT}") |
40 |
| - string(REGEX REPLACE "define NV_TENSORRT_MAJOR * +([0-9]+)" "\\1" NV_TENSORRT_MAJOR "${NV_TENSORRT_MAJOR}") |
41 |
| - string(REGEX MATCH "define NV_TENSORRT_MINOR * +([0-9]+)" NV_TENSORRT_MINOR "${NVINFER_VER_CONTENT}") |
42 |
| - string(REGEX REPLACE "define NV_TENSORRT_MINOR * +([0-9]+)" "\\1" NV_TENSORRT_MINOR "${NV_TENSORRT_MINOR}") |
43 |
| - string(REGEX MATCH "define NV_TENSORRT_PATCH * +([0-9]+)" NV_TENSORRT_PATCH "${NVINFER_VER_CONTENT}") |
44 |
| - string(REGEX REPLACE "define NV_TENSORRT_PATCH * +([0-9]+)" "\\1" NV_TENSORRT_PATCH "${NV_TENSORRT_PATCH}") |
45 |
| - math(EXPR NV_TENSORRT_MAJOR_INT "${NV_TENSORRT_MAJOR}") |
46 |
| - math(EXPR NV_TENSORRT_MINOR_INT "${NV_TENSORRT_MINOR}") |
47 |
| - math(EXPR NV_TENSORRT_PATCH_INT "${NV_TENSORRT_PATCH}") |
48 |
| - |
49 |
| - if (NV_TENSORRT_MAJOR) |
50 |
| - MESSAGE(STATUS "NV_TENSORRT_MAJOR is ${NV_TENSORRT_MAJOR}") |
| 39 | + string(REGEX MATCH "define TRT_MAJOR_RTX * +([0-9]+)" NV_TRT_MAJOR_RTX "${NVINFER_VER_CONTENT}") |
| 40 | + string(REGEX REPLACE "define TRT_MAJOR_RTX * +([0-9]+)" "\\1" NV_TRT_MAJOR_RTX "${NV_TRT_MAJOR_RTX}") |
| 41 | + string(REGEX MATCH "define TRT_MINOR_RTX * +([0-9]+)" NV_TRT_MINOR_RTX "${NVINFER_VER_CONTENT}") |
| 42 | + string(REGEX REPLACE "define TRT_MINOR_RTX * +([0-9]+)" "\\1" NV_TRT_MINOR_RTX "${NV_TRT_MINOR_RTX}") |
| 43 | + math(EXPR NV_TRT_MAJOR_RTX_INT "${NV_TRT_MAJOR_RTX}") |
| 44 | + math(EXPR NV_TRT_MINOR_RTX_INT "${NV_TRT_MINOR_RTX}") |
| 45 | + |
| 46 | + if (NV_TRT_MAJOR_RTX) |
| 47 | + MESSAGE(STATUS "NV_TRT_MAJOR_RTX is ${NV_TRT_MAJOR_RTX}") |
51 | 48 | else()
|
52 |
| - MESSAGE(STATUS "Can't find NV_TENSORRT_MAJOR macro") |
| 49 | + MESSAGE(STATUS "Can't find NV_TRT_MAJOR_RTX macro") |
53 | 50 | endif()
|
54 | 51 |
|
55 |
| - # Check TRT version >= 10.0.1.6 |
56 |
| - if ((NV_TENSORRT_MAJOR_INT GREATER 10) OR |
57 |
| - (NV_TENSORRT_MAJOR_INT EQUAL 10 AND NV_TENSORRT_MINOR_INT GREATER 0) OR |
58 |
| - (NV_TENSORRT_MAJOR_INT EQUAL 10 AND NV_TENSORRT_PATCH_INT GREATER 0)) |
59 |
| - set(TRT_GREATER_OR_EQUAL_TRT_10_GA ON) |
60 |
| - else() |
61 |
| - message( FATAL_ERROR "Only TensorRT 10.x or higher is supported." ) |
62 |
| - endif() |
63 |
| - |
64 |
| - # TensorRT 10 GA onwards, the TensorRT libraries will have major version appended to the end on Windows, |
65 |
| - # for example, nvinfer_10.dll, nvonnxparser_10.dll ... |
66 |
| - if (WIN32 AND TRT_GREATER_OR_EQUAL_TRT_10_GA) |
67 |
| - set(NVINFER_LIB "nvinfer_${NV_TENSORRT_MAJOR}") |
68 |
| - set(PARSER_LIB "nvonnxparser_${NV_TENSORRT_MAJOR}") |
| 52 | + if (WIN32) |
| 53 | + set(NVINFER_LIB "tensorrt_rtx_${NV_TRT_MAJOR_RTX}_${NV_TRT_MINOR_RTX}") |
| 54 | + set(PARSER_LIB "tensorrt_onnxparser_rtx_${NV_TRT_MAJOR_RTX}_${NV_TRT_MINOR_RTX}") |
69 | 55 | endif()
|
70 | 56 |
|
71 | 57 | if (NOT NVINFER_LIB)
|
72 |
| - set(NVINFER_LIB "nvinfer") |
| 58 | + set(NVINFER_LIB "tensorrt_rtx") |
73 | 59 | endif()
|
74 | 60 |
|
75 | 61 | if (NOT PARSER_LIB)
|
76 |
| - set(PARSER_LIB "nvonnxparser") |
| 62 | + set(PARSER_LIB "tensorrt_onnxparser_rtx") |
77 | 63 | endif()
|
78 | 64 |
|
79 | 65 | MESSAGE(STATUS "Looking for ${NVINFER_LIB}")
|
|
100 | 86 | set(TENSORRT_LIBRARY ${TENSORRT_LIBRARY_INFER} ${TENSORRT_LIBRARY_NVONNXPARSER})
|
101 | 87 | MESSAGE(STATUS "Find TensorRT libs at ${TENSORRT_LIBRARY}")
|
102 | 88 | else()
|
103 |
| - if (TRT_GREATER_OR_EQUAL_TRT_10_GA) |
104 |
| - set(ONNX_USE_LITE_PROTO ON) |
105 |
| - endif() |
| 89 | + set(ONNX_USE_LITE_PROTO ON) |
| 90 | + |
106 | 91 | onnxruntime_fetchcontent_declare(
|
107 | 92 | onnx_tensorrt
|
108 | 93 | URL ${DEP_URL_onnx_tensorrt}
|
|
0 commit comments