| @@ -300,6 +300,7 @@ class DavinciModel { | |||||
| return op_list_.at(index); | return op_list_.at(index); | ||||
| } | } | ||||
| void SetGlobalStep(void *global_step) { global_step_addr_ = global_step; } | |||||
| void *GetGlobalStep() const { return global_step_addr_; } | void *GetGlobalStep() const { return global_step_addr_; } | ||||
| // get task info for profiling | // get task info for profiling | ||||
| @@ -204,6 +204,8 @@ Status KnownNodeExecutor::LoadTask(const HybridModel &model, const NodePtr &node | |||||
| davinci_model->SetId(model.GetModelId()); | davinci_model->SetId(model.GetModelId()); | ||||
| davinci_model->SetDumpModelName(model.GetModelName()); | davinci_model->SetDumpModelName(model.GetModelName()); | ||||
| davinci_model->SetOmName(model.GetOmName()); | 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 | // set model id as root node's node id | ||||
| davinci_model->SetSubModelId(node->GetOpDesc()->GetId()); | davinci_model->SetSubModelId(node->GetOpDesc()->GetId()); | ||||
| GELOGD("KnownNodeExecutor::LoadTask node id %ld.", node->GetOpDesc()->GetId()); | GELOGD("KnownNodeExecutor::LoadTask node id %ld.", node->GetOpDesc()->GetId()); | ||||
| @@ -1149,9 +1149,9 @@ domi::Status GenerateSingleOp(const std::string& json_file_path) { | |||||
| if (ret != SUCCESS) { | if (ret != SUCCESS) { | ||||
| DOMI_LOGE("Compile op failed. ge ret = %u, op index = %d", ret, index); | DOMI_LOGE("Compile op failed. ge ret = %u, op index = %d", ret, index); | ||||
| ret = domi::FAILED; | 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; | index += 1; | ||||
| } | } | ||||