run face_landmark without leaky memory

pull/5247/head
zhaodezan 5 years ago
parent 8b5c35210f
commit 73f86c6bb0

@ -108,7 +108,6 @@ kernel::LiteKernel *KernelRegistry::GetKernel(const std::vector<tensor::Tensor *
<< schema::EnumNamePrimitiveType((schema::PrimitiveType)primitive->Type()); << schema::EnumNamePrimitiveType((schema::PrimitiveType)primitive->Type());
return nullptr; return nullptr;
} }
this->op_parameters_.emplace_back(parameter);
auto creator = GetCreator(key); auto creator = GetCreator(key);
if (creator != nullptr) { if (creator != nullptr) {
auto kernel = creator(in_tensors, out_tensors, parameter, ctx, key, primitive); auto kernel = creator(in_tensors, out_tensors, parameter, ctx, key, primitive);
@ -117,10 +116,5 @@ kernel::LiteKernel *KernelRegistry::GetKernel(const std::vector<tensor::Tensor *
return nullptr; return nullptr;
} }
KernelRegistry::~KernelRegistry() { KernelRegistry::~KernelRegistry() {}
for (auto op_parameter : op_parameters_) {
free(op_parameter);
}
op_parameters_.clear();
}
} // namespace mindspore::lite } // namespace mindspore::lite

@ -73,7 +73,9 @@ class LiteKernel {
this->out_kernels_.clear(); this->out_kernels_.clear();
} }
virtual ~LiteKernel() = default; virtual ~LiteKernel() {
free(op_parameter_);
}
virtual int Prepare(); virtual int Prepare();

Loading…
Cancel
Save