File tree Expand file tree Collapse file tree 1 file changed +4
-0
lines changed
python/paddle/distributed/fleet/meta_optimizers Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Original file line number Diff line number Diff line change @@ -233,6 +233,7 @@ def _insert_allreduce_ops(self, ring_id):
233
233
block = self .main_program_list [ring_id - 1 ]['program' ].global_block ()
234
234
origin_block = self .main_program .global_block ()
235
235
grad = None
236
+ processed_param_name = set ()
236
237
for idx , op in reversed (list (enumerate (block .ops ))):
237
238
if is_backward_op (op ) and \
238
239
OP_ROLE_VAR_KEY in op .attr_names :
@@ -242,7 +243,10 @@ def _insert_allreduce_ops(self, ring_id):
242
243
assert len (op_role_var ) % 2 == 0
243
244
offset = idx
244
245
for i in range (0 , len (op_role_var ), 2 ):
246
+ param_name = op_role_var [i ]
245
247
param = block .vars [op_role_var [i ]]
248
+ if param_name in processed_param_name : continue
249
+ processed_param_name .add (param_name )
246
250
grad = block .vars [op_role_var [i + 1 ]]
247
251
origin_param = origin_block .vars [op_role_var [i ]]
248
252
if origin_param .is_distributed :
You can’t perform that action at this time.
0 commit comments