From fc5d059ed5497e674bd36dbebd032af672390ef7 Mon Sep 17 00:00:00 2001 From: Martin Pavella Date: Fri, 3 Oct 2025 16:23:21 +0200 Subject: [PATCH] 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 (cherry picked from commit 5e65da861828ab2f5aff34118df365ede97626f4) --- backends/nxp/tests/test_split_group_convolution.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backends/nxp/tests/test_split_group_convolution.py b/backends/nxp/tests/test_split_group_convolution.py index 21ab1c5b59a..4c9f277e34d 100644 --- a/backends/nxp/tests/test_split_group_convolution.py +++ b/backends/nxp/tests/test_split_group_convolution.py @@ -110,7 +110,7 @@ def test_split_group_convolution__2d(self, _, input_shape: list[int], group: int input_data = torch.randn(input_shape, dtype=torch.float32) out1 = original_module(input_data).detach().numpy() out2 = modified_module(input_data).detach().numpy() - assert np.allclose(out1, out2, atol=2.0e-7) + assert np.allclose(out1, out2, atol=2.0e-7, rtol=1.9e-4) # Make sure the graph can be correctly quantized and lowered to edge. ep = _quantize_and_lower_module(