Skip to content

Commit 7e596ce

Browse files
committed
NXP backend: Adjust unit tests to reflect missing bias in biasless convolutions
1 parent 591184b commit 7e596ce

2 files changed

Lines changed: 7 additions & 9 deletions

File tree

backends/nxp/tests/test_integration.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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

backends/nxp/tests/test_qdq_clustering_conv.py

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff 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"

0 commit comments

Comments
 (0)