Browse Source

Clear memory in singlel op graph after pre build

tags/v1.2.0-rc1
tanghuikang 5 years ago
parent
commit
4faadcfec6
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/ccsrc/backend/session/ascend_session.cc

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

@@ -760,6 +760,7 @@ void AscendSession::BuildOpsInGraph(const GraphId &graph_id, const std::map<AnfN
BuildKernel(kernels);
// Record single op graphs in run_op_graphs_ so that these graphs can be reused in BuildOpImpl
for (const auto &single_op_graph : single_op_graphs) {
RunOpMemoryClear(single_op_graph.first.get());
for (const auto &graph_info : single_op_graph.second) {
run_op_graphs_[graph_info] = single_op_graph.first;
MS_LOG(DEBUG) << "Pre build op finished, graph info: " << graph_info;


Loading…
Cancel
Save