From 91fe55a571c117af766cf6d638209be7e6be1a8e Mon Sep 17 00:00:00 2001 From: wxl Date: Fri, 9 Apr 2021 19:24:22 +0800 Subject: [PATCH] fix data directlly connect netoutput scene --- tests/ut/ge/hybrid/ge_hybrid_unittest.cc | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc index bc706165..c424bdb4 100644 --- a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc +++ b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc @@ -174,9 +174,11 @@ TEST_F(UtestGeHybrid, parse_force_infershape_nodes) { ASSERT_EQ(hybrid_model_builder.ParseForceInfershapeNodes(node, *new_node), SUCCESS); } static ComputeGraphPtr BuildDataDirectConnectGraph() { + const char *kRefIndex = "_parent_node_index"; ge::ut::GraphBuilder builder("subgraph"); auto data = builder.AddNode("Data", "Data", 1, 1); - auto netoutput = builder.AddNode("Netoutput", "Netoutput", 1, 1); + auto netoutput = builder.AddNode("NetOutput", "NetOutput", 1, 1); + (void)AttrUtils::SetInt(netoutput->GetOpDesc()->MutableInputDesc(0), kRefIndex, 0); builder.AddDataEdge(data, 0, netoutput, 0); return builder.GetGraph(); @@ -186,6 +188,7 @@ TEST_F(UtestGeHybrid, data_direct_connect) { auto root_graph = make_shared("root_graph"); OpDescPtr op_desc = CreateOpDesc("PartitionedCall", "PartitionedCall"); auto node = root_graph->AddNode(op_desc); + node->SetOwnerComputeGraph(root_graph); auto sub_graph = BuildDataDirectConnectGraph(); sub_graph->SetParentGraph(root_graph); sub_graph->SetParentNode(node);