!7336 [MSLITE][Develop] fix memory leak bug of arm cpu fp16 op deconv_depthwise

Merge pull request !7336 from yangruoqi713/bugfix

mindspore/lite/src/runtime/kernel/arm/fp16/deconvolution_depthwise_fp16.cc

@@ -53,6 +53,8 @@ int DeconvolutionDepthwiseFp16CPUKernel::InitSlideParam() {
 }
 
 int DeconvolutionDepthwiseFp16CPUKernel::InitBuffer() {
+  if (conv_param_->input_channel_ % C8NUM != 0) {
+    need_align_ = true;
     int C8 = UP_DIV(conv_param_->input_channel_, C8NUM);
     int pack_input_size = conv_param_->input_batch_ * conv_param_->input_h_ * conv_param_->input_w_ * C8NUM * C8;
     packed_input_ = reinterpret_cast<float16_t *>(context_->allocator->Malloc(pack_input_size * sizeof(float16_t)));
@@ -68,6 +70,7 @@ int DeconvolutionDepthwiseFp16CPUKernel::InitBuffer() {
       return RET_ERROR;
     }
     memset(packed_output_, 0, pack_output_size * sizeof(float16_t));
+  }
   return RET_OK;
 }
 

mindspore/lite/src/runtime/kernel/arm/fp32/lstm.cc

@@ -27,6 +27,25 @@ using mindspore::lite::RET_OK;
 using mindspore::schema::PrimitiveType_Lstm;
 
 namespace mindspore::kernel {
+void LstmCPUKernel::FreeTmpBuffer() {
+  if (gate_buffer_ != nullptr) {
+    free(gate_buffer_);
+    gate_buffer_ = nullptr;
+  }
+  if (weight_i_ptr_ != nullptr) {
+    free(weight_i_ptr_);
+    weight_i_ptr_ = nullptr;
+  }
+  if (weight_h_ptr_ != nullptr) {
+    free(weight_h_ptr_);
+    weight_h_ptr_ = nullptr;
+  }
+  if (bias_ptr_ != nullptr) {
+    free(bias_ptr_);
+    bias_ptr_ = nullptr;
+  }
+}
+
 int LstmCPUKernel::InitParam() {
   auto input = in_tensors_.front();
   MS_ASSERT(input != nullptr);

mindspore/lite/src/runtime/kernel/arm/fp32/lstm.h

@@ -37,29 +37,12 @@ class LstmCPUKernel : public LiteKernel {
   int ReSize() override;
   int Run() override;
 
+ private:
+  void FreeTmpBuffer();
   int InitParam();
   int InitBuffer();
   int InitWeightBias();
 
- private:
-  void FreeTmpBuffer() {
-    if (gate_buffer_ != nullptr) {
-      free(gate_buffer_);
-      gate_buffer_ = nullptr;
-    }
-    if (weight_i_ptr_ != nullptr) {
-      free(weight_i_ptr_);
-      weight_i_ptr_ = nullptr;
-    }
-    if (weight_h_ptr_ != nullptr) {
-      free(weight_h_ptr_);
-      weight_h_ptr_ = nullptr;
-    }
-    if (bias_ptr_ != nullptr) {
-      free(bias_ptr_);
-      bias_ptr_ = nullptr;
-    }
-  }
   float *gate_buffer_ = nullptr;
   float *weight_i_ptr_ = nullptr;
   float *weight_h_ptr_ = nullptr;