Browse Source

make manager for every graph

tags/v0.5.0-beta
chenfei 5 years ago
parent
commit
c6bd4ec841
2 changed files with 2 additions and 3 deletions
  1. +1
    -2
      mindspore/ccsrc/device/cpu/cpu_kernel_runtime.cc
  2. +1
    -1
      mindspore/ccsrc/session/session_basic.cc

+ 1
- 2
mindspore/ccsrc/device/cpu/cpu_kernel_runtime.cc View File

@@ -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<ValueNode>()) {
auto value_node = item_node->cast<ValueNodePtr>();


+ 1
- 1
mindspore/ccsrc/session/session_basic.cc View File

@@ -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);


Loading…
Cancel
Save