diff --git a/paddle/fluid/inference/tensorrt/convert/op_converter.h b/paddle/fluid/inference/tensorrt/convert/op_converter.h index da500014b8e819c3a7008a83df2124b5d04560f4..1605490ac452e76f37f85c02bd95e224093bf5b3 100644 --- a/paddle/fluid/inference/tensorrt/convert/op_converter.h +++ b/paddle/fluid/inference/tensorrt/convert/op_converter.h @@ -151,6 +151,13 @@ 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 d109fb051ec3022b474b4a33d1204a38340f6a1c..904768d179d322c1dd57dbab8f17b4b3d16e57b6 100644 --- a/paddle/fluid/inference/tensorrt/op_teller.cc +++ b/paddle/fluid/inference/tensorrt/op_teller.cc @@ -2221,7 +2221,8 @@ struct SimpleOpTypeSetTeller : public Teller { "squeeze2", "unsqueeze2", "layernorm_shift_partition", - "lookup_table"}; + "lookup_table", + "lookup_table_v2"}; std::unordered_set teller_set{ "mul", "matmul", @@ -2333,7 +2334,8 @@ struct SimpleOpTypeSetTeller : public Teller { "unsqueeze2", "fused_token_prune", "layernorm_shift_partition", - "lookup_table"}; + "lookup_table", + "lookup_table_v2"}; }; struct GenericPluginTeller : public Teller {