Browse Source

!9546 Fix PyNative Hccl overflow

From: @jojobugfree
Reviewed-by: @kisnwang,@jjfeing
Signed-off-by: @jjfeing
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
cfa20cceb6
1 changed files with 2 additions and 7 deletions
  1. +2
    -7
      mindspore/ccsrc/backend/kernel_compiler/hccl/hcom_util.cc

+ 2
- 7
mindspore/ccsrc/backend/kernel_compiler/hccl/hcom_util.cc View File

@@ -129,13 +129,8 @@ bool HcomUtil::GetHcomCount(const AnfNodePtr &anf_node, const vector<HcclDataTyp
block_size = input_size / LongToSize(rank_size); block_size = input_size / LongToSize(rank_size);
total_size = total_size + block_size; total_size = total_size + block_size;
} else { } 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 { } else {
block_size = (input_size + align_size - 1 + filled_size) / align_size * align_size; block_size = (input_size + align_size - 1 + filled_size) / align_size * align_size;
} }


Loading…
Cancel
Save