From 4875b9723c12d0489a8ccff6ae65127301a32b37 Mon Sep 17 00:00:00 2001
From: cc <52520497+juncaipeng@users.noreply.github.com>
Date: Tue, 19 Jan 2021 15:08:58 +0800
Subject: [PATCH] Fix bug of supporting channelwise dygraph quantized model,
 test=develop (#30531) (#30545)

---
 .../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