fix trainer_pass.py (#37779) (#37796)

6ece0b1d · zmxdream · GitHub · 56b1ccb7 · 6ece0b1d
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/incubate/fleet/parameter_server/ir/trainer_pass.py .../fluid/incubate/fleet/parameter_server/ir/trainer_pass.py +1 -1

未找到文件。
--- a/python/paddle/fluid/incubate/fleet/parameter_server/ir/trainer_pass.py
+++ b/python/paddle/fluid/incubate/fleet/parameter_server/ir/trainer_pass.py
@@ -534,7 +534,7 @@ def find_heter_ops(program, default_device="cpu"):

                """
                output_vars_no_grad = []
-                for key in pre_op.output_names:
+                for key in op.output_names:
                    for varname in op.output(key):
                        if varname == "@EMPTY@":
                            continue