From 6e415ff7cd1899a09ec35d1aa23c3b73fe0e044d Mon Sep 17 00:00:00 2001 From: wang_shaocong Date: Tue, 6 Apr 2021 15:52:18 +0800 Subject: [PATCH] [MSLITE] fix bug of memory leakage. --- mindspore/lite/nnacl/fp32/conv_depthwise_fp32.c | 1 + mindspore/lite/src/runtime/kernel/arm/base/reshape_base.cc | 1 + 2 files changed, 2 insertions(+) diff --git a/mindspore/lite/nnacl/fp32/conv_depthwise_fp32.c b/mindspore/lite/nnacl/fp32/conv_depthwise_fp32.c index 78580850aa..9aa7e02718 100644 --- a/mindspore/lite/nnacl/fp32/conv_depthwise_fp32.c +++ b/mindspore/lite/nnacl/fp32/conv_depthwise_fp32.c @@ -212,6 +212,7 @@ void ConvDwBorder(float *dst, const float *src, const float *weight, const float param->relu = relu; param->relu6 = relu6; ConvDwFp32Border(param); + free(param); #elif defined(ENABLE_ARM) || defined(ENABLE_SSE) ConvDwFp32Border(dst_kernel, src_kernel, weight_kernel, bias, end_kh - start_kh, end_kw - start_kw, sliding->in_kh_step_ * sizeof(float), sliding->in_kw_step_ * sizeof(float), diff --git a/mindspore/lite/src/runtime/kernel/arm/base/reshape_base.cc b/mindspore/lite/src/runtime/kernel/arm/base/reshape_base.cc index a0a79c9769..65d2b3087b 100644 --- a/mindspore/lite/src/runtime/kernel/arm/base/reshape_base.cc +++ b/mindspore/lite/src/runtime/kernel/arm/base/reshape_base.cc @@ -77,6 +77,7 @@ int ReshapeBaseCPUKernel::Run() { REG_KERNEL(kCPU, kNumberTypeInt32, PrimitiveType_Reshape, LiteKernelCreator) REG_KERNEL(kCPU, kNumberTypeFloat32, PrimitiveType_Reshape, LiteKernelCreator) REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Reshape, LiteKernelCreator) +REG_KERNEL(kCPU, kNumberTypeBool, PrimitiveType_Reshape, LiteKernelCreator) REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Flatten, LiteKernelCreator) REG_KERNEL(kCPU, kNumberTypeFloat32, PrimitiveType_Flatten, LiteKernelCreator) REG_KERNEL(kCPU, kNumberTypeFloat32, PrimitiveType_FlattenGrad, LiteKernelCreator)