Browse Source

add global step info for known subgraph in unknown model and generate om for remained cases when some single op cases run atc failed

tags/v1.5.1
lichun 4 years ago
parent
commit
9530a1631f
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/hybrid/node_executor/compiledsubgraph/known_node_executor.cc

+ 1
- 1
ge/hybrid/node_executor/compiledsubgraph/known_node_executor.cc View File

@@ -188,7 +188,7 @@ void KnownNodeExecutor::SetDaviciModel(const HybridModel &model, const NodePtr &
davinci_model->SetDumpModelName(model.GetModelName());
davinci_model->SetOmName(model.GetOmName());
TensorValue *global_step_var = model.GetVariable(NODE_NAME_GLOBAL_STEP);
GE_CHECK_NOTNULL(global_step_var);
GE_CHK_BOOL_EXEC(global_step_var != nullptr, return);
davinci_model->SetGlobalStep(global_step_var->MutableData());
// set model id as root node's node id
davinci_model->SetSubModelId(node->GetOpDesc()->GetId());


Loading…
Cancel
Save