From d41a56f7f255f2b32434c9ed81a7a5f279b0c795 Mon Sep 17 00:00:00 2001 From: ling Date: Fri, 16 Jul 2021 18:40:46 +0800 Subject: [PATCH] [MSLITE] mindrt actor name unique --- mindspore/lite/src/lite_mindrt.cc | 3 +++ mindspore/lite/src/lite_mindrt.h | 1 + 2 files changed, 4 insertions(+) diff --git a/mindspore/lite/src/lite_mindrt.cc b/mindspore/lite/src/lite_mindrt.cc index 00452088a95..ab4ea575ae5 100644 --- a/mindspore/lite/src/lite_mindrt.cc +++ b/mindspore/lite/src/lite_mindrt.cc @@ -727,6 +727,9 @@ std::vector> CreateOpActor(const std::vectorset_name(kernel->name() + to_string(actor_count++)); + if ((kernel::LiteKernelUtil::IsSwitchCall(kernel))) { auto switch_actor = std::make_shared(kernel); if (switch_actor == nullptr) { diff --git a/mindspore/lite/src/lite_mindrt.h b/mindspore/lite/src/lite_mindrt.h index cb362ab4955..2edd9ce8455 100644 --- a/mindspore/lite/src/lite_mindrt.h +++ b/mindspore/lite/src/lite_mindrt.h @@ -150,6 +150,7 @@ class LiteSwitchOpActor : public LiteOpActor { int MindrtInit(); void MindrtTerminate(const std::vector> &); +static std::atomic_int64_t actor_count = 0; std::vector> CreateOpActor(const std::vector &kernels, const lite::InnerContext *ctx); } // namespace mindspore::lite