未验证 提交 836a662c 编写于 作者: Z zhaoyingli 提交者: GitHub

[AutoParallel] fix process_mesh's member (#49371)

上级 02019804
......@@ -229,8 +229,8 @@ class QuantizationPass(PassBase):
var_dist_attr = dist_context.get_tensor_dist_attr_for_program(var)
dist_attr = {
"dims_mapping": var_dist_attr.dims_mapping,
"process_shape": var_dist_attr.process_mesh.topology,
"process_group": var_dist_attr.process_mesh.processes,
"process_shape": var_dist_attr.process_mesh.shape,
"process_group": var_dist_attr.process_mesh.process_ids,
}
# slice tensor_value with dist_attr
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册