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

recover_skip_quant (#46889)

上级 39548282
...@@ -2409,6 +2409,12 @@ bool OpTeller::Tell(const framework::ir::Node* node, ...@@ -2409,6 +2409,12 @@ bool OpTeller::Tell(const framework::ir::Node* node,
bool with_dynamic_shape) { bool with_dynamic_shape) {
const std::string op_type = node->Op()->Type(); const std::string op_type = node->Op()->Type();
const framework::OpDesc desc = *node->Op(); const framework::OpDesc desc = *node->Op();
// do not support the op which is labeled the `skip_quant`
if ((desc.HasAttr("namescope") &&
PADDLE_GET_CONST(std::string, desc.GetAttr("op_namescope")) ==
"/skip_quant_2/") ||
desc.HasAttr("skip_quant"))
return false;
auto& default_teller = GetDefaultTeller(); auto& default_teller = GetDefaultTeller();
if ((*default_teller)(desc, use_no_calib_int8, with_dynamic_shape)) { if ((*default_teller)(desc, use_no_calib_int8, with_dynamic_shape)) {
SetOpConverterType(op_type, OpConverterType::Default); SetOpConverterType(op_type, OpConverterType::Default);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册