p83571920
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!1828 remove get_rank() in HostAllGather 

Merge pull request !1828 from yihuaijie/master
This commit is contained in:
mindspore-ci-bot 2020-06-04 15:03:07 +08:00 committed by Gitee
parent cd90dfa12e 2041752f33
commit 52c5cb5bdf
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ops/operations/comm_ops.py
View File

@ -218,8 +218,6 @@ class HostAllGather(PrimitiveWithInfer):
validator.check_int_range("rank_id", r, 0, 7, Rel.INC_BOTH, self.name)
validator.check_value_type("rank_id", r, (int,), self.name)
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)
def infer_shape(self, x_shape):