diff --git a/mindspore/lite/CMakeLists.txt b/mindspore/lite/CMakeLists.txt index 679c1db108..e5406d4f0a 100644 --- a/mindspore/lite/CMakeLists.txt +++ b/mindspore/lite/CMakeLists.txt @@ -35,7 +35,7 @@ option(ENABLE_VERBOSE "" off) option(ENABLE_SSE "if x86_64 support SSE instruction set" off) option(ENABLE_AVX "if x86_64 support SSE instruction set" off) option(ENABLE_MINDRT "if support mindrt" on) -option(SUBGRAPH_SPLIT "if support sub graph split" on) +option(SUBGRAPH_SPLIT "if support sub graph split" off) set(DIR_PREFIX mindspore-lite) set(MS_VERSION ${MS_VERSION_MAJOR}.${MS_VERSION_MINOR}.${MS_VERSION_REVISION}) diff --git a/mindspore/lite/src/sub_graph_split.cc b/mindspore/lite/src/sub_graph_split.cc index ee2fb84d95..237c3e8051 100644 --- a/mindspore/lite/src/sub_graph_split.cc +++ b/mindspore/lite/src/sub_graph_split.cc @@ -268,13 +268,18 @@ void SearchSubGraph::SubgraphFusion() { return; } - -void SearchSubGraph::SubGraphSplitByOutput() { - if (!context_->IsGpuEnabled() || output_nodes_.size() > 4) { - return; +bool SearchSubGraph::ModelValid() { + if (context_->IsNpuEnabled()) { + return false; + } + if (context_->IsGpuEnabled()) { + return false; } + return false; +} - if (context_->IsCpuFloat16Enabled() || context_->IsGpuFloat16Enabled()) { +void SearchSubGraph::SubGraphSplitByOutput() { + if (!ModelValid()) { return; } diff --git a/mindspore/lite/src/sub_graph_split.h b/mindspore/lite/src/sub_graph_split.h index 2202293d63..5d9697caf2 100644 --- a/mindspore/lite/src/sub_graph_split.h +++ b/mindspore/lite/src/sub_graph_split.h @@ -63,6 +63,7 @@ class SearchSubGraph { void InitSubgraphDevice(); void SubgraphFusion(); void InitMainGraphDevice(); + bool ModelValid(); private: const InnerContext *context_ = nullptr;