p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!9361 support repeated calculation when the axis of gatherv2 is 0 and split 

From: @yangzhenzhang
Reviewed-by: @kisnwang,@stsuteng
Signed-off-by: @stsuteng
This commit is contained in:
mindspore-ci-bot 2020-12-02 17:18:35 +08:00 committed by Gitee
parent 0c46093496 2f45b7171f
commit 7c4470dc74
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/frontend/parallel/ops_info/gather_v2_p_info.cc
View File

@ -500,6 +500,10 @@ Status GatherV2PInfo::InferBias() {
// params_size=1, axis=0
if ((input_shape.size() == 1) && (axis_ == 0)) {
slice_size_ = input_shape.at(0) / params_strategy.at(0);
// if repeated calculation, because the repeated num in the right of dev-matrix, so rank need to div repeated num
if (repeated_calc_num_ > 1) {
rank = rank / repeated_calc_num_;
}
bias_ = rank * slice_size_;
return SUCCESS;
}

1
mindspore/ccsrc/frontend/parallel/pipeline_transformer/pipeline_transformer.cc
View File

@ -60,7 +60,6 @@ void PipelineTransformer::Coloring() {
}
}
}
return;
}
void PipelineTransformer::BroadCastColoring() {