Browse Source

report error when user forget setting environment variable EXPERIMENTAL_DYNAMIC_PARTITION in dynmaic shape scene

pull/878/head
lichun 5 years ago
parent
commit
86bc06be7e
1 changed files with 6 additions and 0 deletions
  1. +6
    -0
      ge/graph/build/graph_builder.cc

+ 6
- 0
ge/graph/build/graph_builder.cc View File

@@ -200,6 +200,12 @@ Status GraphBuilder::Build(ComputeGraphPtr &comp_graph, std::vector<SubGraphInfo
GeModelPtr ge_model_ptr = nullptr;

if (comp_graph->GetGraphUnknownFlag()) {
if (std::getenv("EXPERIMENTAL_DYNAMIC_PARTITION") == nullptr) {
GELOGE(FAILED, "Environment variable EXPERIMENTAL_DYNAMIC_PARTITION is missing, which is "
"required by current dynamic model[%s], please retry with environment variable "
"\"export EXPERIMENTAL_DYNAMIC_PARTITION=1\"", comp_graph->GetName().c_str());
return FAILED;
}
GE_CHK_STATUS_RET(
BuildForDynamicShapeGraph(comp_graph, subgraph_ptr_list, ge_root_model_ptr, ge_model_ptr, session_id),
"Build for dynamic shape graph failed.");


Loading…
Cancel
Save