From fdf68717473efbd95de8d17d42197d2a01e342b2 Mon Sep 17 00:00:00 2001
From: bukejiyu <395822456@qq.com>
Date: Tue, 9 Jan 2024 09:23:50 +0000
Subject: [PATCH] support trt9.0

---
 paddle/fluid/inference/tensorrt/convert/CMakeLists.txt | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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()