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 3 years ago
parent
commit
95944c1796
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      ge/hybrid/node_executor/compiledsubgraph/known_node_executor.cc

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

@@ -188,6 +188,7 @@ Status KnownNodeExecutor::SetDaviciModel(const HybridModel &model, const NodePtr
davinci_model->SetDumpModelName(model.GetModelName()); davinci_model->SetDumpModelName(model.GetModelName());
davinci_model->SetOmName(model.GetOmName()); davinci_model->SetOmName(model.GetOmName());
void *global_step = model.GetGlobalStep(); void *global_step = model.GetGlobalStep();
GE_CHECK_NOTNULL(global_step);
davinci_model->SetGlobalStep(global_step, sizeof(int64_t)); davinci_model->SetGlobalStep(global_step, sizeof(int64_t));
// set model id as root node's node id // set model id as root node's node id
davinci_model->SetSubModelId(node->GetOpDesc()->GetId()); davinci_model->SetSubModelId(node->GetOpDesc()->GetId());


Loading…
Cancel
Save