From 6a36977d3446d85f1db5da2d7520112159305a89 Mon Sep 17 00:00:00 2001 From: lilong12 Date: Thu, 8 Jul 2021 14:23:46 +0800 Subject: [PATCH] fix the bug, test=develop (#33996) --- python/paddle/fluid/optimizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/optimizer.py b/python/paddle/fluid/optimizer.py index 4fcbdee70e15f..eb2852db188c8 100755 --- a/python/paddle/fluid/optimizer.py +++ b/python/paddle/fluid/optimizer.py @@ -4897,7 +4897,7 @@ def _rename_gradient_var_name(self, block): input_names = op.input_arg_names output_names = op.output_arg_names in_out_names = input_names + output_names - if op.type == 'cast': continue + if op.type == 'cast' or op.type == "c_sync_comm_stream": continue # append "MERGED" to the names of parameter gradients, # and mofify the op_role_var attribute (by rename_arg func). for name in in_out_names: