fix the bug, test=develop (#33996)

PaddlePaddle · Jul 8, 2021 · 6a36977 · 6a36977
1 parent 05643dc
commit 6a36977
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: