From 73f86c6bb04453a9e72a90b5feca11f5badcae3e Mon Sep 17 00:00:00 2001 From: zhaodezan Date: Wed, 26 Aug 2020 04:33:45 -0400 Subject: [PATCH] run face_landmark without leaky memory --- mindspore/lite/src/kernel_registry.cc | 8 +------- mindspore/lite/src/lite_kernel.h | 4 +++- 2 files changed, 4 insertions(+), 8 deletions(-) diff --git a/mindspore/lite/src/kernel_registry.cc b/mindspore/lite/src/kernel_registry.cc index 7396ae78d5..d5a8f1a4d7 100644 --- a/mindspore/lite/src/kernel_registry.cc +++ b/mindspore/lite/src/kernel_registry.cc @@ -108,7 +108,6 @@ kernel::LiteKernel *KernelRegistry::GetKernel(const std::vectorType()); return nullptr; } - this->op_parameters_.emplace_back(parameter); auto creator = GetCreator(key); if (creator != nullptr) { auto kernel = creator(in_tensors, out_tensors, parameter, ctx, key, primitive); @@ -117,10 +116,5 @@ kernel::LiteKernel *KernelRegistry::GetKernel(const std::vectorout_kernels_.clear(); } - virtual ~LiteKernel() = default; + virtual ~LiteKernel() { + free(op_parameter_); + } virtual int Prepare();