From de5dfddd2c4c6fbfe8911c2787fe0e6e3901808a Mon Sep 17 00:00:00 2001 From: zhaozhixuan Date: Mon, 10 May 2021 14:13:48 +0800 Subject: [PATCH] Optimize performance of single_op executor. --- ge/hybrid/model/hybrid_model_builder.cc | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/ge/hybrid/model/hybrid_model_builder.cc b/ge/hybrid/model/hybrid_model_builder.cc index 8ca66a71..a6595df0 100755 --- a/ge/hybrid/model/hybrid_model_builder.cc +++ b/ge/hybrid/model/hybrid_model_builder.cc @@ -127,12 +127,9 @@ Status CollectDependenciesForFusedGraph(NodeItem &node_item, std::set return SUCCESS; } -bool CheckHasHostMem(NodeItem *src_node_item) { - if (src_node_item == nullptr) { - return false; - } - if (src_node_item->NodeType() == DATA) { - auto op_desc = src_node_item->GetOpDesc(); +bool CheckHasHostMem(NodeItem &node_item) { + if (node_item.NodeType() == DATA) { + auto op_desc = node_item->GetOpDesc(); if (op_desc == nullptr) { return false; } @@ -381,7 +378,8 @@ Status HybridModelBuilder::ParseDependentInputNodes(NodeItem &node_item, const s const auto &src_node = peer_out_anchor->GetOwnerNode(); GE_CHECK_NOTNULL(src_node); auto src_node_item = MutableNodeItem(src_node); - GE_IF_BOOL_EXEC(CheckHasHostMem(src_node_item), + GE_CHECK_NOTNULL(src_node_item); + GE_IF_BOOL_EXEC(CheckHasHostMem(*src_node_item), GELOGD("Skip d2h memcpy, get hostmem from node %s.", src_node_item->NodeName().c_str()); continue;) src_node_item->to_const_output_id_list.emplace(peer_out_anchor->GetIdx());