From 10f986a1ee8c16542e1540fb5baf01c9ad3a6bd7 Mon Sep 17 00:00:00 2001 From: zhangxiaokun Date: Thu, 24 Dec 2020 15:45:09 +0800 Subject: [PATCH] return when task sink failed. --- ge/common/helper/model_helper.cc | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/ge/common/helper/model_helper.cc b/ge/common/helper/model_helper.cc index d89c326f..52917abe 100644 --- a/ge/common/helper/model_helper.cc +++ b/ge/common/helper/model_helper.cc @@ -616,11 +616,11 @@ Status ModelHelper::LoadModelData(OmFileLoadHelper &om_load_helper) { } void ModelHelper::SetModelToGeModel(GeModelPtr &ge_model, Model &model) { - model_->SetGraph(model.GetGraph()); - model_->SetName(model.GetName()); - model_->SetVersion(model.GetVersion()); - model_->SetPlatformVersion(model.GetPlatformVersion()); - model_->SetAttr(model.MutableAttrMap()); + ge_model->SetGraph(model.GetGraph()); + ge_model->SetName(model.GetName()); + ge_model->SetVersion(model.GetVersion()); + ge_model->SetPlatformVersion(model.GetPlatformVersion()); + ge_model->SetAttr(model.MutableAttrMap()); } Status ModelHelper::LoadModelData(OmFileLoadHelper &om_load_helper, GeModelPtr &cur_model, size_t mode_index) {