Browse Source

!12069 Change GatherV2P to GatherV2

From: @liangzhibo
Reviewed-by: @ginfung,@zh_qh
Signed-off-by: @zh_qh
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
56650c7b2c
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/rec_generate_strategy.cc

+ 2
- 2
mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/rec_generate_strategy.cc View File

@@ -618,7 +618,7 @@ Dimensions PrepareIncomingOperatorInputStrategy(const std::vector<std::shared_pt
auto name = ops[incoming_op_index]->name().substr(0, pos);
if (name == "Gather") {
return s;
} else if (name == "GatherV2P") {
} else if (name == "GatherP") {
return PrepareGatherV2POutputStrategy(ops, incoming_op_index);
} else {
MS_LOG(EXCEPTION) << "Failure: Unknown type of GatherV2." << std::endl;
@@ -851,7 +851,7 @@ Strategys GenerateStrategiesFromStrategy(const std::vector<std::shared_ptr<Opera
auto name = ops[iter_ops]->name().substr(0, pos);
if (name == "Gather") {
return PrepareGatherV2(ops, iter_ops, basic_stra);
} else if (name == "GatherV2P") {
} else if (name == "GatherP") {
return PrepareGatherV2P(ops, iter_ops, basic_stra);
} else {
MS_LOG(EXCEPTION) << "Failure: Unknown type of GatherV2." << std::endl;


Loading…
Cancel
Save