提交 52c5cb5b 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!1828 remove get_rank() in HostAllGather

Merge pull request !1828 from yihuaijie/master
...@@ -218,8 +218,6 @@ class HostAllGather(PrimitiveWithInfer): ...@@ -218,8 +218,6 @@ class HostAllGather(PrimitiveWithInfer):
validator.check_int_range("rank_id", r, 0, 7, Rel.INC_BOTH, self.name) validator.check_int_range("rank_id", r, 0, 7, Rel.INC_BOTH, self.name)
validator.check_value_type("rank_id", r, (int,), self.name) validator.check_value_type("rank_id", r, (int,), self.name)
self.group_size = len(group) self.group_size = len(group)
self.rank = get_rank()
validator.check('rank', self.rank, 'group', self.group, Rel.IN, self.name)
self.add_prim_attr('group', group) self.add_prim_attr('group', group)
def infer_shape(self, x_shape): def infer_shape(self, x_shape):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册