Browse Source

run face_landmark without leaky memory

tags/v1.0.0
zhaodezan 5 years ago
parent
commit
73f86c6bb0
2 changed files with 4 additions and 8 deletions
  1. +1
    -7
      mindspore/lite/src/kernel_registry.cc
  2. +3
    -1
      mindspore/lite/src/lite_kernel.h

+ 1
- 7
mindspore/lite/src/kernel_registry.cc View File

@@ -108,7 +108,6 @@ kernel::LiteKernel *KernelRegistry::GetKernel(const std::vector<tensor::Tensor *
<< schema::EnumNamePrimitiveType((schema::PrimitiveType)primitive->Type());
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::vector<tensor::Tensor *
return nullptr;
}

KernelRegistry::~KernelRegistry() {
for (auto op_parameter : op_parameters_) {
free(op_parameter);
}
op_parameters_.clear();
}
KernelRegistry::~KernelRegistry() {}
} // namespace mindspore::lite

+ 3
- 1
mindspore/lite/src/lite_kernel.h View File

@@ -73,7 +73,9 @@ class LiteKernel {
this->out_kernels_.clear();
}

virtual ~LiteKernel() = default;
virtual ~LiteKernel() {
free(op_parameter_);
}

virtual int Prepare();



Loading…
Cancel
Save