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;
       }