p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!15509 disable subgraph split 

From: @ling_qiao_min
Reviewed-by: @zhanghaibo5,@zhang_xue_tong
Signed-off-by: @zhang_xue_tong
This commit is contained in:
mindspore-ci-bot 2021-04-23 17:34:26 +08:00 committed by Gitee
parent e3dc9a75cc 6c876edf69
commit 552800e03a
3 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
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_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})

15
mindspore/lite/src/sub_graph_split.cc
View File

@ -268,13 +268,18 @@ void SearchSubGraph::SubgraphFusion() {
return;
}
bool SearchSubGraph::ModelValid() {
if (context_->IsNpuEnabled()) {
return false;
}
if (context_->IsGpuEnabled()) {
return false;
}
return false;
}
void SearchSubGraph::SubGraphSplitByOutput() {
if (!context_->IsGpuEnabled() || output_nodes_.size() > 4) {
return;
}
if (context_->IsCpuFloat16Enabled() || context_->IsGpuFloat16Enabled()) {
if (!ModelValid()) {
return;
}

1
mindspore/lite/src/sub_graph_split.h
View File

@ -63,6 +63,7 @@ class SearchSubGraph {
void InitSubgraphDevice();
void SubgraphFusion();
void InitMainGraphDevice();
bool ModelValid();
private:
const InnerContext *context_ = nullptr;