diff --git a/paddle/fluid/inference/tensorrt/convert/CMakeLists.txt b/paddle/fluid/inference/tensorrt/convert/CMakeLists.txt index 3b304788de606..7cc4201420cb2 100755 --- a/paddle/fluid/inference/tensorrt/convert/CMakeLists.txt +++ b/paddle/fluid/inference/tensorrt/convert/CMakeLists.txt @@ -126,8 +126,9 @@ if(CUSPARSELT_FOUND AND ${TENSORRT_MAJOR_VERSION} GREATER_EQUAL 8) list(APPEND CONVERT_FILES sparse_fc_op.cc sparse_multihead_matmul_op.cc) endif() -if(${TENSORRT_MAJOR_VERSION} GREATER_EQUAL 8 AND ${TENSORRT_MINOR_VERSION} - GREATER_EQUAL 2) +set(TENSORRT_VERSION_NUMBER + "${TENSORRT_MAJOR_VERSION}${TENSORRT_MINOR_VERSION}") +if(${TENSORRT_VERSION_NUMBER} GREATER_EQUAL 82) list(APPEND CONVERT_FILES set_value_op.cc) endif()