Fix bug of supporting channelwise dygraph quantized model, test=develop (#30531)

4 years ago · ce6777fcdf
parent c0fb03a0dc
commit ce6777fcdf
1 changed files with 1 additions and 2 deletions
--- a/python/paddle/fluid/contrib/slim/quantization/quant2_int8_mkldnn_pass.py
+++ b/python/paddle/fluid/contrib/slim/quantization/quant2_int8_mkldnn_pass.py
@ -154,8 +154,7 @@ class Quant2Int8MkldnnPass(object):
                is_quantized_op = True
                for var_node in op_node.inputs:
                    for front_op_node in var_node.inputs:
-                        if "fake_quantize_dequantize_" not in front_op_node.name(
-                        ):
+                        if "quantize_dequantize" not in front_op_node.name():
                            is_quantized_op = False
                if not is_quantized_op:
                    op_node.op()._set_attr("skip_quant", True)