Browse Source

!15509 disable subgraph split

From: @ling_qiao_min
Reviewed-by: @zhanghaibo5,@zhang_xue_tong
Signed-off-by: @zhang_xue_tong
pull/15509/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
552800e03a
3 changed files with 12 additions and 6 deletions
  1. +1
    -1
      mindspore/lite/CMakeLists.txt
  2. +10
    -5
      mindspore/lite/src/sub_graph_split.cc
  3. +1
    -0
      mindspore/lite/src/sub_graph_split.h

+ 1
- 1
mindspore/lite/CMakeLists.txt View File

@@ -35,7 +35,7 @@ option(ENABLE_VERBOSE "" off)
option(ENABLE_SSE "if x86_64 support SSE instruction set" 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_AVX "if x86_64 support SSE instruction set" off)
option(ENABLE_MINDRT "if support mindrt" on) 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(DIR_PREFIX mindspore-lite)
set(MS_VERSION ${MS_VERSION_MAJOR}.${MS_VERSION_MINOR}.${MS_VERSION_REVISION}) set(MS_VERSION ${MS_VERSION_MAJOR}.${MS_VERSION_MINOR}.${MS_VERSION_REVISION})


+ 10
- 5
mindspore/lite/src/sub_graph_split.cc View File

@@ -268,13 +268,18 @@ void SearchSubGraph::SubgraphFusion() {


return; 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; return;
} }




+ 1
- 0
mindspore/lite/src/sub_graph_split.h View File

@@ -63,6 +63,7 @@ class SearchSubGraph {
void InitSubgraphDevice(); void InitSubgraphDevice();
void SubgraphFusion(); void SubgraphFusion();
void InitMainGraphDevice(); void InitMainGraphDevice();
bool ModelValid();


private: private:
const InnerContext *context_ = nullptr; const InnerContext *context_ = nullptr;


Loading…
Cancel
Save