sfqcyy
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!3358 fix cpu nonop net 

Merge pull request !3358 from kisnwang/fix-cpu-nonop-error
This commit is contained in:
mindspore-ci-bot 2020-07-23 17:29:59 +08:00 committed by Gitee
parent 684ff4f46b 2d14147601
commit e4c8365dfe
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
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();
}

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