|
|
@@ -1039,9 +1039,13 @@ Status HybridModelBuilder::InitWeights() { |
|
|
|
GELOGI("Init weight mem successfully, weight base %p, weight size = %zu", |
|
|
|
weight_base, |
|
|
|
sub_weight_buffer->GetSize()); |
|
|
|
auto root_graph = ge_root_model_->GetRootGraph()->GetSubgraph(subgraph_model.first); |
|
|
|
hybrid_model_.weight_buffer_map_.emplace(root_graph->GetName(), std::move(sub_weight_buffer)); |
|
|
|
for (auto &node : root_graph->GetDirectNode()) { |
|
|
|
auto subgraph = GraphUtils::GetComputeGraph(subgraph_model.second->GetGraph()); |
|
|
|
if (subgraph != ge_root_model_->GetRootGraph()) { |
|
|
|
subgraph = ge_root_model_->GetRootGraph()->GetSubgraph(subgraph_model.first); |
|
|
|
} |
|
|
|
GE_CHECK_NOTNULL(subgraph); |
|
|
|
hybrid_model_.weight_buffer_map_.emplace(subgraph->GetName(), std::move(sub_weight_buffer)); |
|
|
|
for (auto &node : subgraph->GetDirectNode()) { |
|
|
|
if (node->GetType() != CONSTANT) { |
|
|
|
continue; |
|
|
|
} |
|
|
|