From ce6777fcdfe2fbbde52a1adff27d36e2de0dcf50 Mon Sep 17 00:00:00 2001 From: cc <52520497+juncaipeng@users.noreply.github.com> Date: Mon, 18 Jan 2021 21:08:02 +0800 Subject: [PATCH] Fix bug of supporting channelwise dygraph quantized model, test=develop (#30531) --- .../fluid/contrib/slim/quantization/quant2_int8_mkldnn_pass.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/python/paddle/fluid/contrib/slim/quantization/quant2_int8_mkldnn_pass.py b/python/paddle/fluid/contrib/slim/quantization/quant2_int8_mkldnn_pass.py index a79b1ee18b1..d93a2059bdc 100644 --- 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) -- GitLab