Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

benchmark mem check fixed

This commit is contained in:
kai00 2020-08-22 14:20:02 +08:00
parent 0c60f7e6ac
commit a9771d63e1
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mindspore/lite/src/lite_session.cc
View File

@ -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;

2
mindspore/lite/tools/benchmark/benchmark.cc
View File

@ -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;