From bad787ed801d8d71d12aa5958e5caf217a295621 Mon Sep 17 00:00:00 2001 From: zmxdream Date: Thu, 2 Dec 2021 11:50:00 +0800 Subject: [PATCH] fix trainer_pass.py (#37779) --- .../fluid/incubate/fleet/parameter_server/ir/trainer_pass.py | 2 +- 1 file changed, 1 insertion(+), 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 index 11fa70b70ba85..ff10c8ea097ad 100644 --- 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