Browse Source

!7475 move reset of reshape tensor_infos to another position

Merge pull request !7475 from yao_yf/reshape_tensorinfos_clear_change_position
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
da7df0dca5
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/frontend/parallel/ops_info/reshape_info.cc

+ 1
- 1
mindspore/ccsrc/frontend/parallel/ops_info/reshape_info.cc View File

@@ -423,7 +423,6 @@ void ReshapeInfo::SetCostForReshape(const mindspore::parallel::StrategyPtr &stra
std::make_shared<StrategyWithCost>(strategy, inputs_tensor_info_, outputs_tensor_info_);
swc->cost_list.push_back(result);
strategy_cost_.emplace_back(swc);
ResetQueueMember();
}

Status ReshapeInfo::GenerateStrategies(int32_t stage_id) {
@@ -489,6 +488,7 @@ Status ReshapeInfo::GenetateStrategyCosts(const std::vector<std::shared_ptr<Stra
}
TensorInfo next_in_tensor_info = next_in_tensor_infos[in_index];
SetOutputLayout(next_in_tensor_info.tensor_layout());
ResetQueueMember();
InferTensorInfoByLayout();
SetCostForReshape(reshape_stra);
}


Loading…
Cancel
Save