未验证 提交 f778c170 编写于 作者: W Wangzheee 提交者: GitHub

[Paddle Inference]disable_lookup_table_v2 (#47515)

* disable_lookup_table_v2
上级 dce81ee1
......@@ -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());
}
......
......@@ -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"};
};
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册