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);
}
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<ComputeGraph>("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);


Loading…
Cancel
Save