From 6ece0b1d4c42ac8c50a3a13af37fdf9882bf0602 Mon Sep 17 00:00:00 2001 From: zmxdream Date: Fri, 3 Dec 2021 19:48:26 +0800 Subject: [PATCH] fix trainer_pass.py (#37779) (#37796) --- .../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 a0832d886d6..bbdc232cc9a 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 @@ -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 -- GitLab