[Paddle-Trt]update set_value cmakelist (#60664)

[Paddle-Trt]update set_value cmakelist
PaddlePaddle · Jan 9, 2024 · a159cd1 · a159cd1
1 parent 114f0b6
commit a159cd1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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()