提交 0f362bb1 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!4980 [MS][LITE][Develop]benchmark mem check fixed

Merge pull request !4980 from wangchangkai/master
......@@ -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;
......
......@@ -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;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册