From 7a212593946333ff22d7bd36cb1df332945367ed Mon Sep 17 00:00:00 2001 From: Wangzheee <634486483@qq.com> Date: Mon, 4 Jul 2022 23:05:37 +0800 Subject: [PATCH] fix delete_quant_dequant_op_pass (#44046) --- paddle/fluid/framework/ir/delete_quant_dequant_op_pass.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/paddle/fluid/framework/ir/delete_quant_dequant_op_pass.cc b/paddle/fluid/framework/ir/delete_quant_dequant_op_pass.cc index 40861638a2..e0d490ce83 100644 --- a/paddle/fluid/framework/ir/delete_quant_dequant_op_pass.cc +++ b/paddle/fluid/framework/ir/delete_quant_dequant_op_pass.cc @@ -89,6 +89,7 @@ void DeleteQuantDequantOpPass::ApplyImpl(ir::Graph* graph) const { std::string quantized_op_type = op_desc->Type(); op_desc->SetAttr("Input_scale", input_scale); op_desc->SetAttr("bit_length", bit_length); + op_desc->SetAttr("enable_int8", true); op_desc->RenameInput(quant_dequant_output_name, input_name); op_desc->Flush(); IR_NODE_LINK_TO(input, quantized_node); -- GitLab