| @@ -127,12 +127,9 @@ Status CollectDependenciesForFusedGraph(NodeItem &node_item, std::set<OpDesc *> | |||||
| return SUCCESS; | 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) { | if (op_desc == nullptr) { | ||||
| return false; | return false; | ||||
| } | } | ||||
| @@ -381,7 +378,8 @@ Status HybridModelBuilder::ParseDependentInputNodes(NodeItem &node_item, const s | |||||
| const auto &src_node = peer_out_anchor->GetOwnerNode(); | const auto &src_node = peer_out_anchor->GetOwnerNode(); | ||||
| GE_CHECK_NOTNULL(src_node); | GE_CHECK_NOTNULL(src_node); | ||||
| auto src_node_item = MutableNodeItem(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()); | GELOGD("Skip d2h memcpy, get hostmem from node %s.", src_node_item->NodeName().c_str()); | ||||
| continue;) | continue;) | ||||
| src_node_item->to_const_output_id_list.emplace(peer_out_anchor->GetIdx()); | src_node_item->to_const_output_id_list.emplace(peer_out_anchor->GetIdx()); | ||||