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