From f778c17007f901c257e9c9ca62a281fad60479dc Mon Sep 17 00:00:00 2001
From: Wangzheee <634486483@qq.com>
Date: Thu, 3 Nov 2022 20:28:43 +0800
Subject: [PATCH] [Paddle Inference]disable_lookup_table_v2 (#47515)

* disable_lookup_table_v2
---
 paddle/fluid/inference/tensorrt/convert/op_converter.h | 7 -------
 paddle/fluid/inference/tensorrt/op_teller.cc           | 4 ++--
 2 files changed, 2 insertions(+), 9 deletions(-)

diff --git a/paddle/fluid/inference/tensorrt/convert/op_converter.h b/paddle/fluid/inference/tensorrt/convert/op_converter.h
index 2dd0e4656f..74074303b9 100644
--- a/paddle/fluid/inference/tensorrt/convert/op_converter.h
+++ b/paddle/fluid/inference/tensorrt/convert/op_converter.h
@@ -151,13 +151,6 @@ class OpConverter {
               platform::errors::Unimplemented("no OpConverter for optype [%s]",
                                               op_desc.Type()));
         }
-        if (op_desc.Type() == "lookup_table_v2") {
-          it = Registry<OpConverter>::Global().Lookup("lookup_table");
-          PADDLE_ENFORCE_NOT_NULL(
-              it,
-              platform::errors::Unimplemented("no OpConverter for optype [%s]",
-                                              op_desc.Type()));
-        }
         if (!it) {
           it = Registry<OpConverter>::Global().Lookup(op_desc.Type());
         }
diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc
index 85aa5e82d7..f6901d4d78 100644
--- a/paddle/fluid/inference/tensorrt/op_teller.cc
+++ b/paddle/fluid/inference/tensorrt/op_teller.cc
@@ -2274,7 +2274,7 @@ struct SimpleOpTypeSetTeller : public Teller {
       "logsigmoid",
       "preln_layernorm_shift_partition",
       "lookup_table",
-      "lookup_table_v2",
+      // "lookup_table_v2",
       "expand_v2"};
 
   std::unordered_set<std::string> teller_set{
@@ -2397,7 +2397,7 @@ struct SimpleOpTypeSetTeller : public Teller {
       "preln_layernorm_shift_partition",
       "merge_layernorm",
       "lookup_table",
-      "lookup_table_v2",
+      // "lookup_table_v2",
       "expand_v2"};
 };
 
-- 
GitLab