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

!6521 [Auto parallel] Checking the return value of find() in rec-searching 

Merge pull request !6521 from Xiaoda/24-fix-a-return-value-checking-bug-in-rec-searching
This commit is contained in:
mindspore-ci-bot 2020-09-19 18:22:51 +08:00 committed by Gitee
parent 85df184347 56d610f681
commit 05d5ac9096
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/rec_generate_strategy.cc
View File

@ -487,6 +487,9 @@ Dimensions PrepareIncomingOperatorInputStrategy(const std::vector<std::shared_pt
}
if (ops[incoming_op_index]->type() == GATHERV2) {
auto pos = ops[incoming_op_index]->name().find("Info");
if (pos == std::string::npos) {
return s;
}
auto name = ops[incoming_op_index]->name().substr(0, pos);
if (name == "GatherV2") {
return s;