Browse Source

!10358 [AutoParallel] UnsortedSegmentMax op strategy adjustement

From: @ch-l
Reviewed-by: @zhunaipan,@stsuteng
Signed-off-by: @stsuteng
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
98a3f318c7
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/rec_generate_strategy.cc

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

@@ -370,6 +370,9 @@ Strategys MakeRecSearchStrategy(const std::shared_ptr<Graph> &graph,
if (iter_ops >= ops.size()) {
MS_LOG(EXCEPTION) << "Failure: Operators' elements out of range.";
}
if (graph->nodes[iter_graph].apply.op_type == kRecUnsortedSegmentOp) {
return MakeDataParallelStrategy(graph, ops, iter_graph, iter_ops);
}

StrategyPtr origin_strategy = ops[iter_ops]->strategy();
Strategys strategies;


Loading…
Cancel
Save