Browse Source

!13248 mindrt memory leak

From: @ling_qiao_min
Reviewed-by: @zhanghaibo5,@zhang_xue_tong,@zhanghaibo5
Signed-off-by: @zhang_xue_tong
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
88a3b36fa8
3 changed files with 10 additions and 1 deletions
  1. +7
    -0
      mindspore/lite/src/lite_mindrt.cc
  2. +2
    -0
      mindspore/lite/src/lite_mindrt.h
  3. +1
    -1
      mindspore/lite/src/mindrt_executor.h

+ 7
- 0
mindspore/lite/src/lite_mindrt.cc View File

@@ -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) {


+ 2
- 0
mindspore/lite/src/lite_mindrt.h View File

@@ -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


+ 1
- 1
mindspore/lite/src/mindrt_executor.h View File

@@ -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