p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!5451 remove parameter destruction in subgraph 

Merge pull request !5451 from wandongdong/master
This commit is contained in:
mindspore-ci-bot 2020-08-28 22:00:22 +08:00 committed by Gitee
parent fd82e4f85a 04314fdc85
commit ad638d69de
2 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/lite/src/lite_kernel.h
View File

@ -74,7 +74,10 @@ class LiteKernel {
} }
virtual ~LiteKernel() { virtual ~LiteKernel() {
free(op_parameter_); if (op_parameter_ != nullptr) {
free(op_parameter_);
op_parameter_ = nullptr;
}
} }
virtual int Prepare(); virtual int Prepare();

11
mindspore/lite/src/runtime/kernel/opencl/subgraph_opencl_kernel.cc
View File

@ -261,26 +261,21 @@ int SubGraphOpenCLKernel::GetKernelFromToTensor(const std::vector<lite::tensor::
} }
int SubGraphOpenCLKernel::UnInit() { int SubGraphOpenCLKernel::UnInit() {
for (const auto tensor : in_convert_tensors_) { for (const auto &tensor : in_convert_tensors_) {
if (tensor != nullptr) { if (tensor != nullptr) {
delete tensor; delete tensor;
} }
} }
for (const auto tensor : out_convert_tensors_) { for (const auto &tensor : out_convert_tensors_) {
if (tensor != nullptr) { if (tensor != nullptr) {
delete tensor; delete tensor;
} }
} }
for (const auto op : in_convert_ops_) { for (const auto &op : in_convert_ops_) {
if (op != nullptr) { if (op != nullptr) {
delete op; delete op;
} }
} }
for (const auto parameter : in_parameters_) {
if (parameter != nullptr) {
delete parameter;
}
}
return RET_OK; return RET_OK;
} }