Browse Source

!1828 remove get_rank() in HostAllGather

Merge pull request !1828 from yihuaijie/master
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
52c5cb5bdf
1 changed files with 0 additions and 2 deletions
  1. +0
    -2
      mindspore/ops/operations/comm_ops.py

+ 0
- 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):


Loading…
Cancel
Save