Browse Source

disable subgraph split

pull/15509/head
ling 4 years ago
parent
commit
6c876edf69
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_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})


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

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



+ 1
- 0
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;


Loading…
Cancel
Save