m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix bug in BuildOpsInGraph

Browse Source
pull/12032/head
tanghuikang 4 years ago
parent b2fb825d89
commit b136c5f0eb
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File

4
mindspore/ccsrc/backend/session/ascend_session.cc
Unescape View File

@ -703,7 +703,7 @@ void AscendSession::GetOpInputStubTensors(const CNodePtr &cnode, const std::map<
void AscendSession::BuildOpsInGraph(const GraphId &graph_id, const std::map<AnfNodePtr, size_t> &parameter_index, void AscendSession::BuildOpsInGraph(const GraphId &graph_id, const std::map<AnfNodePtr, size_t> &parameter_index,
const std::vector<tensor::TensorPtr> &graph_inputs) { const std::vector<tensor::TensorPtr> &graph_inputs) {
if (built_graph_id_.find(graph_id) == built_graph_id_.end()) { if (built_graph_id_.find(graph_id) != built_graph_id_.end()) {
return; return;
} }
auto graph = GetGraph(graph_id); auto graph = GetGraph(graph_id);
@ -753,7 +753,7 @@ void AscendSession::BuildOpsInGraph(const GraphId &graph_id, const std::map<AnfN
for (const auto &single_op_graph : single_op_graphs) { for (const auto &single_op_graph : single_op_graphs) {
for (const auto &graph_info : single_op_graph.second) { for (const auto &graph_info : single_op_graph.second) {
run_op_graphs_[graph_info] = single_op_graph.first; run_op_graphs_[graph_info] = single_op_graph.first;
MS_LOG(DEBUG) << "Pre build op finished, graph info: " << single_op_graph.second; MS_LOG(DEBUG) << "Pre build op finished, graph info: " << graph_info;
} }
} }
built_graph_id_.insert(graph_id); built_graph_id_.insert(graph_id);

Write Preview
Loading…
Cancel
Save