From 70ecf3b12da42748b6c881976897189f1f1709ce Mon Sep 17 00:00:00 2001
From: wenbin <wang3323032@qq.com>
Date: Mon, 5 Jul 2021 20:14:46 +0800
Subject: [PATCH] correct define (#33966)

* correct op

* test=allcase
---
 paddle/fluid/inference/tensorrt/op_teller.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc
index 59b196e3d92..c21ef8840de 100644
--- a/paddle/fluid/inference/tensorrt/op_teller.cc
+++ b/paddle/fluid/inference/tensorrt/op_teller.cc
@@ -51,7 +51,7 @@ struct SimpleOpTypeSetTeller : public Teller {
 #if IS_TRT_VERSION_GE(7130)
     teller_set.insert("group_norm");
 #endif
-#if CUDA_VERSION >= 10200
+#if CUDA_VERSION >= 10020
     teller_set.insert("reshape");
     teller_set.insert("reshape2");
 #endif
@@ -719,6 +719,8 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8,
 
     if ((*teller)(op_type, desc, use_no_calib_int8)) return true;
   }
+
+  VLOG(3) << "trt unsupported op " << op_type;
   return false;
 }
 
-- 
GitLab