Browse Source

fix

tags/v1.2.0
wjm 4 years ago
parent
commit
48d7b6dc8b
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/hybrid/model/hybrid_model_builder.cc

+ 1
- 1
ge/hybrid/model/hybrid_model_builder.cc View File

@@ -154,7 +154,7 @@ Status HybridModelBuilder::Build() {

Status HybridModelBuilder::BuildForSingleOp() {
GE_CHK_STATUS_RET(ValidateParams(), "Failed to validate GeRootModel");
hybrid_model_.model_name_ = ge_root_model_->GetModelName();
hybrid_model_.model_name_ = ge_root_model_->GetRootGraph()->GetName();
GELOGI("[%s] Start to build hybrid model.", GetGraphName());
auto ret = ge_root_model_->GetSubgraphInstanceNameToModel();
const GeModelPtr ge_model = ret[ge_root_model_->GetRootGraph()->GetName()];


Loading…
Cancel
Save