Browse Source

fix data directlly connect netoutput scene

tags/v1.3.0
wxl 3 years ago
parent
commit
91fe55a571
1 changed files with 4 additions and 1 deletions
  1. +4
    -1
      tests/ut/ge/hybrid/ge_hybrid_unittest.cc

+ 4
- 1
tests/ut/ge/hybrid/ge_hybrid_unittest.cc View File

@@ -174,9 +174,11 @@ TEST_F(UtestGeHybrid, parse_force_infershape_nodes) {
ASSERT_EQ(hybrid_model_builder.ParseForceInfershapeNodes(node, *new_node), SUCCESS); ASSERT_EQ(hybrid_model_builder.ParseForceInfershapeNodes(node, *new_node), SUCCESS);
} }
static ComputeGraphPtr BuildDataDirectConnectGraph() { static ComputeGraphPtr BuildDataDirectConnectGraph() {
const char *kRefIndex = "_parent_node_index";
ge::ut::GraphBuilder builder("subgraph"); ge::ut::GraphBuilder builder("subgraph");
auto data = builder.AddNode("Data", "Data", 1, 1); 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); builder.AddDataEdge(data, 0, netoutput, 0);
return builder.GetGraph(); return builder.GetGraph();
@@ -186,6 +188,7 @@ TEST_F(UtestGeHybrid, data_direct_connect) {
auto root_graph = make_shared<ComputeGraph>("root_graph"); auto root_graph = make_shared<ComputeGraph>("root_graph");
OpDescPtr op_desc = CreateOpDesc("PartitionedCall", "PartitionedCall"); OpDescPtr op_desc = CreateOpDesc("PartitionedCall", "PartitionedCall");
auto node = root_graph->AddNode(op_desc); auto node = root_graph->AddNode(op_desc);
node->SetOwnerComputeGraph(root_graph);
auto sub_graph = BuildDataDirectConnectGraph(); auto sub_graph = BuildDataDirectConnectGraph();
sub_graph->SetParentGraph(root_graph); sub_graph->SetParentGraph(root_graph);
sub_graph->SetParentNode(node); sub_graph->SetParentNode(node);


Loading…
Cancel
Save