Skip to content

Commit ca9fc06

Browse files
[Release Only] Bugfix/fix nxp separable conv test (#14800)
### Summary Fix failing separable convolution test. The error is larger on the CI than on my PC. Fixes #14709 ### Test plan N/A Co-authored-by: Martin Pavella <[email protected]>
1 parent 0b5a4ab commit ca9fc06

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
@@ -110,7 +110,7 @@ def test_split_group_convolution__2d(self, _, input_shape: list[int], group: int
110110
input_data = torch.randn(input_shape, dtype=torch.float32)
111111
out1 = original_module(input_data).detach().numpy()
112112
out2 = modified_module(input_data).detach().numpy()
113-
assert np.allclose(out1, out2, atol=2.0e-7)
113+
assert np.allclose(out1, out2, atol=2.0e-7, rtol=1.9e-4)
114114

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

0 commit comments

Comments
 (0)