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

checking the find() return value

This commit is contained in:
Xiaoda Zhang 2020-09-19 09:46:30 +08:00
parent ba2fe29691
commit 56d610f681
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;