未验证 提交 d1da885f 编写于 作者: X xu98bin 提交者: GitHub

solve auto_aprallel pp2 with fp16 question (#49913)

上级 6b151c0e
......@@ -1850,7 +1850,7 @@ class Completer:
op_dist_attr.set_output_dims_mapping(
input_var.name, ref_dims_mapping
)
if "SkipUpdate" not in input_name:
input_var_attr.process_mesh = ref_process_mesh
self._dist_context.set_tensor_dist_attr_for_program(
input_var, input_var_attr
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册