|
|
@@ -395,24 +395,6 @@ Status GraphBuilder::BuildForHostCpuGraph(ComputeGraphPtr &comp_graph, GeModelPt |
|
|
|
return BuildForUnknownShapeGraph(comp_graph, ge_model_ptr, session_id); |
|
|
|
} |
|
|
|
|
|
|
|
static Status InsertMemcpyNode(const ComputeGraphPtr &graph, const OutDataAnchorPtr &out_anchor, |
|
|
|
const std::vector<InDataAnchorPtr> &in_anchors, const std::string &name) { |
|
|
|
GE_CHECK_NOTNULL(out_anchor); |
|
|
|
NodePtr in_node = out_anchor->GetOwnerNode(); |
|
|
|
GE_CHECK_NOTNULL(in_node); |
|
|
|
OpDescBuilder op_desc_builder(name, MEMCPYASYNC); |
|
|
|
OpDescPtr op_desc = op_desc_builder.AddInput("x", in_node->GetOpDesc()->GetOutputDesc(0)) |
|
|
|
.AddOutput("y", in_node->GetOpDesc()->GetOutputDesc(0)) |
|
|
|
.Build(); |
|
|
|
(void)AttrUtils::SetBool(op_desc, ATTR_NO_NEED_CONSTANT_FOLDING, false); |
|
|
|
if (GraphUtils::InsertNodeAfter(out_anchor, in_anchors, graph->AddNode(op_desc)) != GRAPH_SUCCESS) { |
|
|
|
REPORT_CALL_ERROR("E19999", "Insert IDENTITY node %s after %s failed", name.c_str(), in_node->GetName().c_str()); |
|
|
|
GELOGE(FAILED, "Insert IDENTITY node %s after %s failed.", name.c_str(), in_node->GetName().c_str()); |
|
|
|
return FAILED; |
|
|
|
} |
|
|
|
return SUCCESS; |
|
|
|
} |
|
|
|
|
|
|
|
Status GraphBuilder::MarkFpBpProfilingTaskAttr(ComputeGraphPtr &com_graph) { |
|
|
|
bool original_unknown_shape_flag = com_graph->GetGraphUnknownFlag(); |
|
|
|
com_graph->SetGraphUnknownFlag(false); |
|
|
|