提交 04314fdc 编写于 作者: W wandongdong

remove parameter destory in subgraph

上级 5ffb9588
......@@ -74,7 +74,10 @@ class LiteKernel {
}
virtual ~LiteKernel() {
free(op_parameter_);
if (op_parameter_ != nullptr) {
free(op_parameter_);
op_parameter_ = nullptr;
}
}
virtual int Prepare();
......
......@@ -261,26 +261,21 @@ int SubGraphOpenCLKernel::GetKernelFromToTensor(const std::vector<lite::tensor::
}
int SubGraphOpenCLKernel::UnInit() {
for (const auto tensor : in_convert_tensors_) {
for (const auto &tensor : in_convert_tensors_) {
if (tensor != nullptr) {
delete tensor;
}
}
for (const auto tensor : out_convert_tensors_) {
for (const auto &tensor : out_convert_tensors_) {
if (tensor != nullptr) {
delete tensor;
}
}
for (const auto op : in_convert_ops_) {
for (const auto &op : in_convert_ops_) {
if (op != nullptr) {
delete op;
}
}
for (const auto parameter : in_parameters_) {
if (parameter != nullptr) {
delete parameter;
}
}
return RET_OK;
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册