fix trainer_pass.py (PaddlePaddle#37779)

Zjq9409 · Dec 10, 2021 · bad787e · bad787e
1 parent d947f64
commit bad787e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 _append_heter_op(op, current_heter_block_ops, heter_ops):
 
                 """
                 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