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

pull/1902/head
lichun 4 years ago
parent
commit
421ad5b8fa
3 changed files with 5 additions and 2 deletions
  1. +1
    -0
      ge/graph/load/model_manager/davinci_model.h
  2. +2
    -0
      ge/hybrid/node_executor/compiledsubgraph/known_node_executor.cc
  3. +2
    -2
      ge/offline/main.cc

+ 1
- 0
ge/graph/load/model_manager/davinci_model.h View File

@@ -300,6 +300,7 @@ class DavinciModel {
return op_list_.at(index);
}

void SetGlobalStep(void *global_step) { global_step_addr_ = global_step; }
void *GetGlobalStep() const { return global_step_addr_; }

// get task info for profiling


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

@@ -204,6 +204,8 @@ Status KnownNodeExecutor::LoadTask(const HybridModel &model, const NodePtr &node
davinci_model->SetId(model.GetModelId());
davinci_model->SetDumpModelName(model.GetModelName());
davinci_model->SetOmName(model.GetOmName());
TensorValue *global_step_var = model.GetVariable(NODE_NAME_GLOBAL_STEP);
davinci_model->SetGlobalStep(global_step_var->MutableData());
// set model id as root node's node id
davinci_model->SetSubModelId(node->GetOpDesc()->GetId());
GELOGD("KnownNodeExecutor::LoadTask node id %ld.", node->GetOpDesc()->GetId());


+ 2
- 2
ge/offline/main.cc View File

@@ -1149,9 +1149,9 @@ domi::Status GenerateSingleOp(const std::string& json_file_path) {
if (ret != SUCCESS) {
DOMI_LOGE("Compile op failed. ge ret = %u, op index = %d", ret, index);
ret = domi::FAILED;
break;
} else {
GELOGI("Compile op success. op index = %d, output = %s", index, output_path.c_str());
}
GELOGI("Compile op success. op index = %d, output = %s", index, output_path.c_str());
index += 1;
}



Loading…
Cancel
Save