fix QuantWeightPass error in sub graph (#50203)

26a8b5b2 · Guanghua Yu · GitHub · d09962a1 · 26a8b5b2
显示空白变更内容
内联并排

Showing with 9 addition and 5 deletion

python/paddle/static/quantization/quantization_pass.py python/paddle/static/quantization/quantization_pass.py +9 -5

未找到文件。
--- a/python/paddle/static/quantization/quantization_pass.py
+++ b/python/paddle/static/quantization/quantization_pass.py
@@ -3226,6 +3226,10 @@ class QuantWeightPass:
                    self._quantized_ops[x_node.name()] = quant_weight_node

                for next_op_node in out_node.outputs:
+                    if (
+                        self._quantized_ops[x_node.name()].node
+                        in graph.graph.nodes()
+                    ):
                        graph.update_input_link(
                            out_node,
                            self._quantized_ops[x_node.name()],