forked from mindspore-Ecosystem/mindspore
fix coredump on cpu
This commit is contained in:
parent
20a1a7d54e
commit
b0a2a74bf8
|
@ -225,7 +225,7 @@ void CPUKernelRuntime::CreateOutputTensors(session::KernelGraph *kernel_graph,
|
|||
if (input_nodes.size() != inputs.size()) {
|
||||
MS_LOG(EXCEPTION) << "Input size not equal to input node size!";
|
||||
}
|
||||
input_param_tensor_map_.clear();
|
||||
|
||||
size_t input_idx = 0;
|
||||
for (auto &item : input_nodes) {
|
||||
MS_EXCEPTION_IF_NULL(item);
|
||||
|
@ -240,6 +240,7 @@ void CPUKernelRuntime::CreateOutputTensors(session::KernelGraph *kernel_graph,
|
|||
auto out = CreatTensorForOutput(kernel_graph, item_with_index, tensor_to_node);
|
||||
outputs->push_back(std::move(out));
|
||||
}
|
||||
input_param_tensor_map_.clear();
|
||||
}
|
||||
|
||||
void CPUKernelRuntime::BindInputTensorAddressPtr(const session::KernelGraph &kernel_graph,
|
||||
|
|
Loading…
Reference in New Issue