fix trainer_pass.py (#37779)

f6d485a8 · zmxdream · GitHub · 67cb5476 · f6d485a8
隐藏空白更改
内联并排

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
@@ -771,7 +771,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