mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!19988 [MS][LITE][Develop]fix bug of memleak of control flow model 

Merge pull request !19988 from mengyuanli/bugfix
This commit is contained in:
i-robot 2021-07-11 10:27:32 +00:00 committed by Gitee
parent ac06df9640 cabb53a05e
commit 9df1092f5c
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mindspore/lite/src/lite_mindrt.h
View File

@ -53,6 +53,8 @@ class LiteOpActor : public OpActor<lite::Tensor> {
isolate_input_tensor->set_data(nullptr);
delete isolate_input_tensor;
}
delete call_node_;
delete partial_node_;
}
void RunOpData(OpData<lite::Tensor> *input_data, OpContext<lite::Tensor> *context = nullptr) override;
virtual int CompileArrow();
@ -109,7 +111,12 @@ class LiteOpActor : public OpActor<lite::Tensor> {
class LiteSwitchOpActor : public LiteOpActor {
public:
explicit LiteSwitchOpActor(kernel::LiteKernel *kernel) : LiteOpActor(kernel) {}
~LiteSwitchOpActor() override = default;
~LiteSwitchOpActor() override {
delete call_node_;
delete switch_node_;
delete true_partial_node_;
delete false_partial_node_;
};
void RunOpData(OpData<Tensor> *inputs, OpContext<Tensor> *context = nullptr) override;
int CompileArrow() override;
int PrepareOutputData() override;