diff --git a/paddle/fluid/inference/tensorrt/convert/op_converter.h b/paddle/fluid/inference/tensorrt/convert/op_converter.h index 2dd0e4656f2ffebe29bfd879075edc89a82c34e1..74074303b923ab84ecf5d580aaa061924c292757 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::Global().Lookup("lookup_table"); - PADDLE_ENFORCE_NOT_NULL( - it, - platform::errors::Unimplemented("no OpConverter for optype [%s]", - op_desc.Type())); - } if (!it) { it = Registry::Global().Lookup(op_desc.Type()); } diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc index 85aa5e82d7b5f038f70382d82824ca36bceecffc..f6901d4d783be77183ff85aa8480d943f9450121 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 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"}; };