From a21ef87e6e859812df1cd28cfc64fc04e339e7ae Mon Sep 17 00:00:00 2001
From: caifubi <caifubi1@huawei.com>
Date: Sat, 5 Dec 2020 13:48:11 +0800
Subject: [PATCH] fix pynative hccl overflow

---
 .../ccsrc/backend/kernel_compiler/hccl/hcom_util.cc      | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/mindspore/ccsrc/backend/kernel_compiler/hccl/hcom_util.cc b/mindspore/ccsrc/backend/kernel_compiler/hccl/hcom_util.cc
index 9aa6dcf0e6..bf88fef013 100644
--- a/mindspore/ccsrc/backend/kernel_compiler/hccl/hcom_util.cc
+++ b/mindspore/ccsrc/backend/kernel_compiler/hccl/hcom_util.cc
@@ -129,13 +129,8 @@ bool HcomUtil::GetHcomCount(const AnfNodePtr &anf_node, const vector<HcclDataTyp
       block_size = input_size / LongToSize(rank_size);
       total_size = total_size + block_size;
     } else {
-      if (AnfAlgo::GetCNodeName(anf_node) == kAllGatherOpName) {
-        auto cnode = anf_node->cast<CNodePtr>();
-        if (AnfAlgo::HasNodeAttr(kAttrFusion, cnode) && AnfAlgo::GetNodeAttr<int64_t>(anf_node, kAttrFusion)) {
-          block_size = (input_size + align_size - 1 + filled_size) / align_size * align_size;
-        } else {
-          block_size = input_size;
-        }
+      if (i == size - 1) {
+        block_size = input_size;
       } else {
         block_size = (input_size + align_size - 1 + filled_size) / align_size * align_size;
       }