Browse Source

[MSLITE] fix bug of memory leakage.

pull/14664/head
wang_shaocong 4 years ago
parent
commit
6e415ff7cd
2 changed files with 2 additions and 0 deletions
  1. +1
    -0
      mindspore/lite/nnacl/fp32/conv_depthwise_fp32.c
  2. +1
    -0
      mindspore/lite/src/runtime/kernel/arm/base/reshape_base.cc

+ 1
- 0
mindspore/lite/nnacl/fp32/conv_depthwise_fp32.c View File

@@ -212,6 +212,7 @@ void ConvDwBorder(float *dst, const float *src, const float *weight, const float
param->relu = relu; param->relu = relu;
param->relu6 = relu6; param->relu6 = relu6;
ConvDwFp32Border(param); ConvDwFp32Border(param);
free(param);
#elif defined(ENABLE_ARM) || defined(ENABLE_SSE) #elif defined(ENABLE_ARM) || defined(ENABLE_SSE)
ConvDwFp32Border(dst_kernel, src_kernel, weight_kernel, bias, end_kh - start_kh, end_kw - start_kw, 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), sliding->in_kh_step_ * sizeof(float), sliding->in_kw_step_ * sizeof(float),


+ 1
- 0
mindspore/lite/src/runtime/kernel/arm/base/reshape_base.cc View File

@@ -77,6 +77,7 @@ int ReshapeBaseCPUKernel::Run() {
REG_KERNEL(kCPU, kNumberTypeInt32, PrimitiveType_Reshape, LiteKernelCreator<ReshapeBaseCPUKernel>) REG_KERNEL(kCPU, kNumberTypeInt32, PrimitiveType_Reshape, LiteKernelCreator<ReshapeBaseCPUKernel>)
REG_KERNEL(kCPU, kNumberTypeFloat32, PrimitiveType_Reshape, LiteKernelCreator<ReshapeBaseCPUKernel>) REG_KERNEL(kCPU, kNumberTypeFloat32, PrimitiveType_Reshape, LiteKernelCreator<ReshapeBaseCPUKernel>)
REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Reshape, LiteKernelCreator<ReshapeBaseCPUKernel>) REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Reshape, LiteKernelCreator<ReshapeBaseCPUKernel>)
REG_KERNEL(kCPU, kNumberTypeBool, PrimitiveType_Reshape, LiteKernelCreator<ReshapeBaseCPUKernel>)
REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Flatten, LiteKernelCreator<ReshapeBaseCPUKernel>) REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Flatten, LiteKernelCreator<ReshapeBaseCPUKernel>)
REG_KERNEL(kCPU, kNumberTypeFloat32, PrimitiveType_Flatten, LiteKernelCreator<ReshapeBaseCPUKernel>) REG_KERNEL(kCPU, kNumberTypeFloat32, PrimitiveType_Flatten, LiteKernelCreator<ReshapeBaseCPUKernel>)
REG_KERNEL(kCPU, kNumberTypeFloat32, PrimitiveType_FlattenGrad, LiteKernelCreator<ReshapeBaseCPUKernel>) REG_KERNEL(kCPU, kNumberTypeFloat32, PrimitiveType_FlattenGrad, LiteKernelCreator<ReshapeBaseCPUKernel>)


Loading…
Cancel
Save