hummingbird
/
graphengine
Not watched
2
0
Fork 0
Code
Releases 13 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
Browse Source

Pre Merge pull request !1844 from HW_KK/fix-build-for-single-op

pull/1844/MERGE
HW_KK Gitee 3 years ago
parent
460406cbd6 b1b98f17c9
commit
3395bc5e5b
1 changed files with 1 additions and 0 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -0
      ge/hybrid/model/hybrid_model_builder.cc

+ 1
- 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)
: hybrid_model_(hybrid_model), runtime_param_(hybrid_model.root_runtime_param_) {
ge_root_model_ = hybrid_model_.ge_root_model_;
hybrid_model_.root_graph_ = ge_root_model_->GetRootGraph();
}

Status HybridModelBuilder::Build() {


Write Preview
Loading…
Cancel
Save