Browse Source

Optimize performance of single_op executor.

tags/v1.3.0
zhaozhixuan 4 years ago
parent
commit
de5dfddd2c
1 changed files with 5 additions and 7 deletions
  1. +5
    -7
      ge/hybrid/model/hybrid_model_builder.cc

+ 5
- 7
ge/hybrid/model/hybrid_model_builder.cc View File

@@ -127,12 +127,9 @@ Status CollectDependenciesForFusedGraph(NodeItem &node_item, std::set<OpDesc *>
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());


Loading…
Cancel
Save