From a9771d63e13000311e1c6207e02f42971fc16b36 Mon Sep 17 00:00:00 2001
From: kai00 <wangchangkai@huawei.com>
Date: Sat, 22 Aug 2020 14:20:02 +0800
Subject: [PATCH] benchmark mem check fixed

---
 mindspore/lite/src/lite_session.cc          | 7 +++++++
 mindspore/lite/tools/benchmark/benchmark.cc | 2 ++
 2 files changed, 9 insertions(+)

diff --git a/mindspore/lite/src/lite_session.cc b/mindspore/lite/src/lite_session.cc
index 4b8946ace..bae52d709 100644
--- a/mindspore/lite/src/lite_session.cc
+++ b/mindspore/lite/src/lite_session.cc
@@ -287,6 +287,13 @@ LiteSession::~LiteSession() {
   for (auto *kernel : kernels_) {
     delete kernel;
   }
+  for (auto *ms_tensor : input_vec_) {
+    if (ms_tensor != nullptr) {
+      ((tensor::LiteTensor *)ms_tensor)->SetTensorImpl(nullptr);
+      delete ms_tensor;
+    }
+  }
+  input_vec_.clear();
   delete this->context_;
   delete this->executor;
   this->executor = nullptr;
diff --git a/mindspore/lite/tools/benchmark/benchmark.cc b/mindspore/lite/tools/benchmark/benchmark.cc
index 5fdb5e8b4..ada79393e 100644
--- a/mindspore/lite/tools/benchmark/benchmark.cc
+++ b/mindspore/lite/tools/benchmark/benchmark.cc
@@ -102,10 +102,12 @@ int Benchmark::ReadInputFile() {
       if (size != tensorDataSize) {
         std::cerr << "Input binary file size error, required: %zu, in fact: %zu" << tensorDataSize << size << std::endl;
         MS_LOG(ERROR) << "Input binary file size error, required: " << tensorDataSize << ", in fact: " << size;
+        delete binBuf;
         return RET_ERROR;
       }
       auto inputData = cur_tensor->MutableData();
       memcpy(inputData, binBuf, tensorDataSize);
+      delete binBuf;
     }
   }
   return RET_OK;
-- 
GitLab