|
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