Browse Source

!3358 fix cpu nonop net

Merge pull request !3358 from kisnwang/fix-cpu-nonop-error
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
e4c8365dfe
2 changed files with 6 additions and 2 deletions
  1. +5
    -1
      mindspore/ccsrc/runtime/device/cpu/cpu_kernel_runtime.cc
  2. +1
    -1
      mindspore/ccsrc/runtime/device/cpu/cpu_simple_mem_plan.cc

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

@@ -186,11 +186,15 @@ BaseRef CPUKernelRuntime::CreatTensorForOutput(const session::KernelWithIndex &k
return ret;
}
return CreatTensorForOutput(node, index, bound_addresses, need_sync_outputs);
} else if (input_node->isa<Parameter>() || input_node->isa<ValueNode>()) {
} else if (input_node->isa<Parameter>()) {
auto iter = input_map.find(input_node.get());
if (iter != input_map.end()) {
return iter->second;
}
} else if (input_node->isa<ValueNode>()) {
auto value_node = input_node->cast<ValueNodePtr>();
MS_EXCEPTION_IF_NULL(value_node);
return value_node->value();
}
return BaseRef();
}


+ 1
- 1
mindspore/ccsrc/runtime/device/cpu/cpu_simple_mem_plan.cc View File

@@ -21,7 +21,7 @@ namespace device {
namespace cpu {
void CPUSimpleMemPlan::MemPlan(const session::KernelGraph *graph) {
MS_EXCEPTION_IF_NULL(graph);
size_t total_mem_size = 0;
size_t total_mem_size = 32;
auto kernels = graph->execution_order();
for (const auto &kernel : kernels) {
MS_EXCEPTION_IF_NULL(kernel);


Loading…
Cancel
Save