Skip to content

Commit 5e65da8

Browse files
Bugfix/fix nxp separable conv test (#14719)
### Summary Fix failing separable convolution test. The error is larger on the CI than on my PC. Fixes #14709 ### Test plan N/A
1 parent ac5d645 commit 5e65da8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

backends/nxp/tests/test_split_group_convolution.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +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-
assert np.allclose(out1, out2, atol=2.0e-7)
109+
assert np.allclose(out1, out2, atol=2.0e-7, rtol=1.9e-4)
110110

111111
# Make sure the graph can be correctly quantized and lowered to edge.
112112
ep = _quantize_and_lower_module(

0 commit comments

Comments
 (0)