Skip to content

Commit 1d326ff

Browse files
committed
NXP Backend: Fix failing separable convolution test.
1 parent b0462ae commit 1d326ff

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

backends/nxp/tests/test_split_group_convolution.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,7 @@ def test_split_group_convolution__2d(self, _, input_shape: list[int], group: int
106106
input_data = torch.randn(input_shape, dtype=torch.float32)
107107
out1 = original_module(input_data).detach().numpy()
108108
out2 = modified_module(input_data).detach().numpy()
109+
print(np.abs(out1 - out2).max())
109110
assert np.allclose(out1, out2, atol=2.0e-7)
110111

111112
# Make sure the graph can be correctly quantized and lowered to edge.

0 commit comments

Comments
 (0)