Browse Source

dynamic shape attribute setting in hetero devices

tags/v1.1.0
hwjiaorui 5 years ago
parent
commit
81bdf61126
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/backend/session/ascend_session.cc

+ 3
- 0
mindspore/ccsrc/backend/session/ascend_session.cc View File

@@ -121,6 +121,9 @@ GraphId AscendSession::CompileGraphImpl(const AnfNodePtrList &lst, const AnfNode
MS_LOG(INFO) << "Start"; MS_LOG(INFO) << "Start";
// construct graph, if successfully, graph_sum_ + 1 // construct graph, if successfully, graph_sum_ + 1
auto graph = ConstructKernelGraph(lst, outputs); auto graph = ConstructKernelGraph(lst, outputs);
// Update Graph Dynamic Shape Attr
UpdateGraphDynamicShapeAttr(NOT_NULL(graph));
graph->UpdateGraphDynamicAttr();
auto graph_id = graph->graph_id(); auto graph_id = graph->graph_id();
MS_LOG(INFO) << "Compile graph " << graph_id << " success"; MS_LOG(INFO) << "Compile graph " << graph_id << " success";
return graph_id; return graph_id;


Loading…
Cancel
Save