From c6bd4ec841a0dcbcc3ccbb7090d83434fdeb7bfa Mon Sep 17 00:00:00 2001 From: chenfei Date: Wed, 27 May 2020 10:32:23 +0800 Subject: [PATCH] make manager for every graph --- mindspore/ccsrc/device/cpu/cpu_kernel_runtime.cc | 3 +-- mindspore/ccsrc/session/session_basic.cc | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/mindspore/ccsrc/device/cpu/cpu_kernel_runtime.cc b/mindspore/ccsrc/device/cpu/cpu_kernel_runtime.cc index 1dcddfa994..67328f04c2 100644 --- a/mindspore/ccsrc/device/cpu/cpu_kernel_runtime.cc +++ b/mindspore/ccsrc/device/cpu/cpu_kernel_runtime.cc @@ -44,8 +44,7 @@ void CPUKernelRuntime::AssignKernelAddress(session::KernelGraph *kernel_graph) { void CPUKernelRuntime::AssignValueNodeAddress(session::KernelGraph *kernel_graph) { MS_EXCEPTION_IF_NULL(kernel_graph); size_t type_size = sizeof(float); - for (auto &item : kernel_graph->value_nodes()) { - auto item_node = item.first; + for (auto &item_node : kernel_graph->graph_value_nodes()) { MS_EXCEPTION_IF_NULL(item_node); if (item_node->isa()) { auto value_node = item_node->cast(); diff --git a/mindspore/ccsrc/session/session_basic.cc b/mindspore/ccsrc/session/session_basic.cc index db6257c815..15a66c63ab 100644 --- a/mindspore/ccsrc/session/session_basic.cc +++ b/mindspore/ccsrc/session/session_basic.cc @@ -589,7 +589,7 @@ KernelGraphPtr SessionBasic::ConstructKernelGraph(const AnfNodePtrList &lst, con // add a make_tuple at the end of graph as output graph->set_output(ConstructOutput(outputs, graph)); MS_EXCEPTION_IF_NULL(context_); - FuncGraphManagerPtr manager = context_->manager(); + FuncGraphManagerPtr manager = MakeManager({graph}); if (manager) { manager->AddFuncGraph(graph); graph->set_manager(manager);