xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!18727 fix get allreduce fusion bug 

Merge pull request !18727 from kisnwang/fix-hccl-allreduce-fusion-get-bug
This commit is contained in:
i-robot 2021-06-23 02:12:08 +00:00 committed by Gitee
parent d4aca69981 3fbd8ce783
commit a7ddd10af5
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/frontend/parallel/context.cc
View File

@ -149,9 +149,8 @@ void ParallelContext::SetAllReduceFusionSplitIndices(const std::vector<uint32_t>
all_reduce_fusion_split_indices_[group + TypeIdLabel(kNumberTypeFloat)] = indices;
all_reduce_fusion_split_indices_[group + TypeIdLabel(kNumberTypeFloat16)] = indices;
all_reduce_fusion_split_indices_[group + TypeIdLabel(kNumberTypeFloat32)] = indices;
} else {
all_reduce_fusion_split_indices_[group] = indices;
}
all_reduce_fusion_split_indices_[group] = indices;
}
std::vector<uint32_t> ParallelContext::GetAllReduceFusionSplitIndices(const std::string &group) const {
@ -169,9 +168,8 @@ void ParallelContext::SetAllReduceFusionSplitSizes(const std::vector<uint32_t> &
all_reduce_fusion_split_sizes_[group + TypeIdLabel(kNumberTypeFloat)] = sizes;
all_reduce_fusion_split_sizes_[group + TypeIdLabel(kNumberTypeFloat16)] = sizes;
all_reduce_fusion_split_sizes_[group + TypeIdLabel(kNumberTypeFloat32)] = sizes;
} else {
all_reduce_fusion_split_sizes_[group] = sizes;
}
all_reduce_fusion_split_sizes_[group] = sizes;
}
std::vector<uint32_t> ParallelContext::GetAllReduceFusionSplitSizes(const std::string &group) const {