Browse Source

!2569 [CT][ME][parallel] One-hot runs failed in RP-search mode

Merge pull request !2569 from Chong/zc
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
a32d674cc5
2 changed files with 1 additions and 3 deletions
  1. +0
    -2
      mindspore/ccsrc/parallel/auto_parallel/rec_core/rec_generate_strategy.cc
  2. +1
    -1
      mindspore/ccsrc/parallel/auto_parallel/rec_core/rec_parse_graph.cc

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

@@ -370,8 +370,6 @@ std::vector<int32_t> CopyIncomingOperatorOutputStrategy(const std::shared_ptr<Gr
s.push_back(1 / graph->nodes[iter_graph].tensor_parm.tensor_str.str_c);
s.push_back(1 / graph->nodes[iter_graph].tensor_parm.tensor_str.str_h);
s.push_back(1 / graph->nodes[iter_graph].tensor_parm.tensor_str.str_w);
} else if (input_stra_dim == 0) {
s = {};
} else {
MS_LOG(EXCEPTION) << ops[iter_ops]->name() << ": Tensor's shape is unknown.";
}


+ 1
- 1
mindspore/ccsrc/parallel/auto_parallel/rec_core/rec_parse_graph.cc View File

@@ -215,7 +215,7 @@ std::shared_ptr<Graph> EliminateGraph(const std::shared_ptr<Graph> &graph,
const std::shared_ptr<std::vector<std::vector<size_t>>> &eli_list,
const std::shared_ptr<std::vector<size_t>> &index_list) {
MS_EXCEPTION_IF_NULL(graph);
const std::set<OperatorType> elementwise_type = {
static const std::set<OperatorType> elementwise_type = {
OperatorType::kRecReLU, OperatorType::kRecLog, OperatorType::kRecExp, OperatorType::kRecAdd,
OperatorType::kRecElmWiseOp, OperatorType::kRecBiasAdd, OperatorType::kRecSub, OperatorType::kRecMul,
OperatorType::kRecDiv, OperatorType::kRecSqueeze, OperatorType::kRecReduce, OperatorType::kRecCast,


Loading…
Cancel
Save