From 2041752f33fe3fd25fde772b1f527ca2de24e760 Mon Sep 17 00:00:00 2001 From: Yi Huaijie Date: Wed, 3 Jun 2020 20:27:05 +0800 Subject: [PATCH] remove get_rank() op in HostAllGather --- mindspore/ops/operations/comm_ops.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/mindspore/ops/operations/comm_ops.py b/mindspore/ops/operations/comm_ops.py index 6e0c22f584..6fee406a35 100644 --- a/mindspore/ops/operations/comm_ops.py +++ b/mindspore/ops/operations/comm_ops.py @@ -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):