fix quant pass (#49123)

332aadc0 · ceci3 · GitHub · be4ea174 · 332aadc0
隐藏空白更改
内联并排

Showing with 7 addition and 4 deletion

python/paddle/fluid/contrib/slim/quantization/quantization_pass.py ...ddle/fluid/contrib/slim/quantization/quantization_pass.py +7 -4

未找到文件。
--- a/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py
+++ b/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py
@@ -1776,10 +1776,13 @@ class OutScaleForInferencePass:
                    in_node = graph._find_node_by_name(
                        op_node.outputs, var_name
                    )
-                    if in_node.dtype() not in [
+                    if (in_node.node.var() is None) or (
-                        core.VarDesc.VarType.FP64,
+                        in_node.dtype()
-                        core.VarDesc.VarType.FP32,
+                        not in [
-                    ]:
+                            core.VarDesc.VarType.FP64,
+                            core.VarDesc.VarType.FP32,
+                        ]
+                    ):
                        continue
                    scale_name = self._scale_name(var_name)