@@ -47,11 +47,23 @@ if(TENSORRT_FOUND)
47
47
file (READ ${TENSORRT_INCLUDE_DIR} /NvInfer.h TENSORRT_VERSION_FILE_CONTENTS )
48
48
string (REGEX MATCH "define NV_TENSORRT_MAJOR +([0-9]+)" TENSORRT_MAJOR_VERSION
49
49
"${TENSORRT_VERSION_FILE_CONTENTS} " )
50
+ string (REGEX MATCH "define NV_TENSORRT_MINOR +([0-9]+)" TENSORRT_MINOR_VERSION
51
+ "${TENSORRT_VERSION_FILE_CONTENTS} " )
52
+ string (REGEX MATCH "define NV_TENSORRT_PATCH +([0-9]+)" TENSORRT_PATCH_VERSION
53
+ "${TENSORRT_VERSION_FILE_CONTENTS} " )
54
+ string (REGEX MATCH "define NV_TENSORRT_BUILD +([0-9]+)" TENSORRT_BUILD_VERSION
55
+ "${TENSORRT_VERSION_FILE_CONTENTS} " )
50
56
51
57
if ("${TENSORRT_MAJOR_VERSION} " STREQUAL "" )
52
58
file (READ ${TENSORRT_INCLUDE_DIR} /NvInferVersion.h TENSORRT_VERSION_FILE_CONTENTS )
53
59
string (REGEX MATCH "define NV_TENSORRT_MAJOR +([0-9]+)" TENSORRT_MAJOR_VERSION
54
60
"${TENSORRT_VERSION_FILE_CONTENTS} " )
61
+ string (REGEX MATCH "define NV_TENSORRT_MINOR +([0-9]+)" TENSORRT_MINOR_VERSION
62
+ "${TENSORRT_VERSION_FILE_CONTENTS} " )
63
+ string (REGEX MATCH "define NV_TENSORRT_PATCH +([0-9]+)" TENSORRT_PATCH_VERSION
64
+ "${TENSORRT_VERSION_FILE_CONTENTS} " )
65
+ string (REGEX MATCH "define NV_TENSORRT_BUILD +([0-9]+)" TENSORRT_BUILD_VERSION
66
+ "${TENSORRT_VERSION_FILE_CONTENTS} " )
55
67
endif ()
56
68
57
69
if ("${TENSORRT_MAJOR_VERSION} " STREQUAL "" )
@@ -60,9 +72,15 @@ if(TENSORRT_FOUND)
60
72
61
73
string (REGEX REPLACE "define NV_TENSORRT_MAJOR +([0-9]+)" "\\ 1"
62
74
TENSORRT_MAJOR_VERSION "${TENSORRT_MAJOR_VERSION} " )
75
+ string (REGEX REPLACE "define NV_TENSORRT_MINOR +([0-9]+)" "\\ 1"
76
+ TENSORRT_MINOR_VERSION "${TENSORRT_MINOR_VERSION} " )
77
+ string (REGEX REPLACE "define NV_TENSORRT_PATCH +([0-9]+)" "\\ 1"
78
+ TENSORRT_PATCH_VERSION "${TENSORRT_PATCH_VERSION} " )
79
+ string (REGEX REPLACE "define NV_TENSORRT_BUILD +([0-9]+)" "\\ 1"
80
+ TENSORRT_BUILD_VERSION "${TENSORRT_BUILD_VERSION} " )
63
81
64
82
message (STATUS "Current TensorRT header is ${TENSORRT_INCLUDE_DIR} /NvInfer.h. "
65
- "Current TensorRT version is v${TENSORRT_MAJOR_VERSION} . " )
83
+ "Current TensorRT version is v${TENSORRT_MAJOR_VERSION} .${TENSORRT_MINOR_VERSION} . ${TENSORRT_PATCH_VERSION} . ${TENSORRT_BUILD_VERSION} " )
66
84
include_directories (${TENSORRT_INCLUDE_DIR} )
67
85
link_directories (${TENSORRT_LIBRARY} )
68
86
add_definitions (-DPADDLE_WITH_TENSORRT )
0 commit comments