Browse Source

fix build for single op

pull/1844/head
wuweikang 4 years ago
parent
commit
b1b98f17c9
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      ge/hybrid/model/hybrid_model_builder.cc

+ 2
- 0
ge/hybrid/model/hybrid_model_builder.cc View File

@@ -141,6 +141,7 @@ Status CollectDependenciesForFusedGraph(NodeItem &node_item, std::set<OpDesc *>
HybridModelBuilder::HybridModelBuilder(HybridModel &hybrid_model) HybridModelBuilder::HybridModelBuilder(HybridModel &hybrid_model)
: hybrid_model_(hybrid_model), runtime_param_(hybrid_model.root_runtime_param_) { : hybrid_model_(hybrid_model), runtime_param_(hybrid_model.root_runtime_param_) {
ge_root_model_ = hybrid_model_.ge_root_model_; ge_root_model_ = hybrid_model_.ge_root_model_;
hybrid_model_.root_graph_ = ge_root_model_->GetRootGraph();
} }


Status HybridModelBuilder::Build() { Status HybridModelBuilder::Build() {
@@ -852,6 +853,7 @@ Status HybridModelBuilder::LoadGraph() {
root_graph->GetAllNodesSize()); root_graph->GetAllNodesSize());
} }


hybrid_model_.root_graph_ = root_graph;
GE_CHK_STATUS_RET(RelinkNextIteration(), "[%s] Relink NextIteration failed", GetGraphName()); GE_CHK_STATUS_RET(RelinkNextIteration(), "[%s] Relink NextIteration failed", GetGraphName());
// Reset node id by topological order across all subgraphs // Reset node id by topological order across all subgraphs
int64_t index = 0; int64_t index = 0;


Loading…
Cancel
Save