未验证 提交 6a36977d 编写于 作者: L lilong12 提交者: GitHub

fix the bug, test=develop (#33996)

上级 05643dc3
...@@ -4897,7 +4897,7 @@ class PipelineOptimizer(object): ...@@ -4897,7 +4897,7 @@ class PipelineOptimizer(object):
input_names = op.input_arg_names input_names = op.input_arg_names
output_names = op.output_arg_names output_names = op.output_arg_names
in_out_names = input_names + output_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, # append "MERGED" to the names of parameter gradients,
# and mofify the op_role_var attribute (by rename_arg func). # and mofify the op_role_var attribute (by rename_arg func).
for name in in_out_names: for name in in_out_names:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册