| @@ -61,6 +61,13 @@ void LiteOpActor::SetOutputData(OpContext<Tensor> *context) { | |||||
| int MindrtInit() { return mindspore::Initialize("tcp://127.0.0.1:8080", "", "", "", 1); } | 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>> CreateOpActor(const std::vector<kernel::LiteKernel *> &kernels) { | ||||
| std::vector<std::shared_ptr<LiteOpActor>> actors; | std::vector<std::shared_ptr<LiteOpActor>> actors; | ||||
| for (auto kernel : kernels) { | for (auto kernel : kernels) { | ||||
| @@ -88,6 +88,8 @@ class LiteOpActor : public OpActor<lite::Tensor> { | |||||
| }; | }; | ||||
| int MindrtInit(); | int MindrtInit(); | ||||
| void MindrtTerminate(std::vector<std::shared_ptr<LiteOpActor>>); | |||||
| std::vector<std::shared_ptr<LiteOpActor>> CreateOpActor(const std::vector<kernel::LiteKernel *> &kernels); | std::vector<std::shared_ptr<LiteOpActor>> CreateOpActor(const std::vector<kernel::LiteKernel *> &kernels); | ||||
| } // namespace lite | } // namespace lite | ||||
| @@ -29,7 +29,7 @@ namespace mindspore::lite { | |||||
| class MindrtExecutor : public Executor { | class MindrtExecutor : public Executor { | ||||
| public: | public: | ||||
| MindrtExecutor() = default; | MindrtExecutor() = default; | ||||
| virtual ~MindrtExecutor() = default; | |||||
| virtual ~MindrtExecutor() { MindrtTerminate(opActors_); } | |||||
| virtual int Prepare(const std::vector<kernel::LiteKernel *> &kernels); | virtual int Prepare(const std::vector<kernel::LiteKernel *> &kernels); | ||||