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

!9945 fix bit unpack may cause memory leak 

From: @xutianchun
Reviewed-by: @hangangqiang,@HilbertDavid,@HilbertDavid
Signed-off-by: @hangangqiang,@HilbertDavid
This commit is contained in:
mindspore-ci-bot 2020-12-16 10:30:46 +08:00 committed by Gitee
parent a7f99fc752 124a7a4b3a
commit e22e66fa8d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -99,7 +99,6 @@ int LiteSession::ConvertTensorsData(const lite::Model *model, size_t tensor_inde
return RET_ERROR;
}
} else {
MS_ASSERT(dst_tensor->Size() == src_tensor->data()->size());
if (WeightTensorNeedCopy(model, tensor_index)) {
auto dst_data = dst_tensor->MutableData();
if (dst_data == nullptr) {
@ -118,6 +117,7 @@ int LiteSession::ConvertTensorsData(const lite::Model *model, size_t tensor_inde
return RET_ERROR;
}
kernel::DequantUtil::UnPackToInt(src_tensor, dst_tensor->MutableData());
copyed_tensor_idxes_.emplace_back(tensor_index);
} else {
dst_tensor->set_data(const_cast<unsigned char *>(src_tensor->data()->data()));
}