!13248 mindrt memory leak

From: @ling_qiao_min
Reviewed-by: @zhanghaibo5,@zhang_xue_tong,@zhanghaibo5
Signed-off-by: @zhang_xue_tong
pull/13248/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 88a3b36fa8

@ -61,6 +61,13 @@ void LiteOpActor::SetOutputData(OpContext<Tensor> *context) {
int MindrtInit() { return mindspore::Initialize("tcp://127.0.0.1:8080", "", "", "", 1); }
void MindrtTerminate(std::vector<std::shared_ptr<LiteOpActor>> actor_list) {
for (auto actor : actor_list) {
mindspore::Terminate(actor->GetAID());
}
return;
}
std::vector<std::shared_ptr<LiteOpActor>> CreateOpActor(const std::vector<kernel::LiteKernel *> &kernels) {
std::vector<std::shared_ptr<LiteOpActor>> actors;
for (auto kernel : kernels) {

@ -88,6 +88,8 @@ class LiteOpActor : public OpActor<lite::Tensor> {
};
int MindrtInit();
void MindrtTerminate(std::vector<std::shared_ptr<LiteOpActor>>);
std::vector<std::shared_ptr<LiteOpActor>> CreateOpActor(const std::vector<kernel::LiteKernel *> &kernels);
} // namespace lite

@ -29,7 +29,7 @@ namespace mindspore::lite {
class MindrtExecutor : public Executor {
public:
MindrtExecutor() = default;
virtual ~MindrtExecutor() = default;
virtual ~MindrtExecutor() { MindrtTerminate(opActors_); }
virtual int Prepare(const std::vector<kernel::LiteKernel *> &kernels);

Loading…
Cancel
Save