File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -29,7 +29,7 @@ def test_conv_fc_softmax__to_executorch_program(use_qat):
2929 delegation_info = get_delegation_info (program .graph_module )
3030 assert delegation_info .num_delegated_subgraphs == 1
3131 assert delegation_info .num_non_delegated_nodes == 11
32- assert delegation_info .num_delegated_nodes == 14
32+ assert delegation_info .num_delegated_nodes == 13
3333
3434 for node in program .graph .nodes :
3535 # Make sure Convolution and AddMM are delegated
Original file line number Diff line number Diff line change @@ -16,18 +16,16 @@ def test_conv2d_partitioner():
1616 lowered_module = edge_program .exported_program ().graph_module .lowered_module_0
1717 nodes = list (lowered_module .original_module .graph .nodes )
1818
19- assert len (nodes ) == 13
19+ assert len (nodes ) == 9
2020
21- q_x_node = nodes [6 ]
22- dq_w_node = nodes [7 ]
23- dq_x_node = nodes [8 ]
24- dq_bias_node = nodes [9 ]
25- conv_node = nodes [10 ]
26- q_y_node = nodes [11 ]
21+ q_x_node = nodes [3 ]
22+ dq_x_node = nodes [4 ]
23+ dq_w_node = nodes [5 ]
24+ conv_node = nodes [6 ]
25+ q_y_node = nodes [7 ]
2726
2827 assert "cluster" not in q_x_node .meta
2928 assert dq_w_node .meta ["cluster" ] == "aten_convolution_default_cluster"
3029 assert dq_x_node .meta ["cluster" ] == "aten_convolution_default_cluster"
31- assert dq_bias_node .meta ["cluster" ] == "aten_convolution_default_cluster"
3230 assert conv_node .meta ["cluster" ] == "aten_convolution_default_cluster"
3331 assert q_y_node .meta ["cluster" ] == "aten_convolution_default_cluster"
You can’t perform that action at this time.
0 commit comments