xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!13568 [MS][LITE] fix memory leak 

From: @probiotics_53
Reviewed-by: @hangangqiang,@zhang_xue_tong
Signed-off-by: @hangangqiang
This commit is contained in:
mindspore-ci-bot 2021-03-19 09:23:22 +08:00 committed by Gitee
parent a7c1b6a1ef 4aed10f552
commit e8cb23e35e
3 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/lite/tools/converter/legacy_optimizer/graph/infershape_pass.cc
View File

@ -116,7 +116,7 @@ std::vector<Tensor *> ConvertTensorToLiteTensor(MetaGraphT *graph, const std::ve
return lite_tensors;
}
STATUS NodeInferShpae(const std::unique_ptr<schema::CNodeT> &node, const std::vector<Tensor *> &inputs,
STATUS NodeInferShape(const std::unique_ptr<schema::CNodeT> &node, const std::vector<Tensor *> &inputs,
std::vector<Tensor *> *outputs) {
flatbuffers::FlatBufferBuilder fbb(INITIAL_SIZE);
auto prim = ConvertToPrimitive(node->primitive.get(), &fbb);
@ -197,7 +197,7 @@ STATUS InferShapePass::Run(MetaGraphT *graph) {
FreeTensors(input_tensors, output_tensors);
return RET_INFER_ERR;
}
auto status = NodeInferShpae(node, input_tensors, &output_tensors);
auto status = NodeInferShape(node, input_tensors, &output_tensors);
MS_LOG(DEBUG) << "cur node:" << node->name;
if (status == RET_INFER_INVALID) {
MS_LOG(INFO) << "InferShape shouldn't be done before runtime, name: " << node->name

1
mindspore/lite/tools/optimizer/fusion/constant_folding_fusion.cc
View File

@ -121,6 +121,7 @@ kernel::LiteKernel *GetLiteKernel(std::vector<Tensor *> inputs, std::vector<Tens
auto prim_t = lite::GetPrimitiveT(cnode->input(0));
flatbuffers::FlatBufferBuilder fbb(INITIAL_SIZE);
auto prim = lite::ConvertToPrimitive(prim_t, &fbb);
delete prim_t;
if (prim == nullptr) {
fbb.Clear();
MS_LOG(ERROR) << "get primitive failed.";

1
mindspore/lite/tools/optimizer/graph/infershape_pass.cc
View File

@ -397,6 +397,7 @@ bool InferShapePass::Run(const FuncGraphPtr &func_graph) {
flatbuffers::FlatBufferBuilder fbb(INITIAL_SIZE);
auto prim = lite::ConvertToPrimitive(prim_t, &fbb);
delete prim_t;
if (prim == nullptr) {
MS_LOG(ERROR) << "get primitive failed.";
FreeTensors(&input_tensors);