fix test tol

This commit is contained in:
Awni Hannun 2025-08-21 14:45:35 -07:00
parent 3ce23755ea
commit 80868ee4fb

View File

@ -341,7 +341,7 @@ class TestQuantized(mlx_tests.MLXTestCase):
)
y_hat = x @ w_hat
self.assertEqual(y_q.shape, y_hat.shape)
self.assertLess((y_q - y_hat).abs().max(), 1e-3)
self.assertLess((y_q - y_hat).abs().max(), 2e-3)
def test_mode_error_cases(self):
w = mx.random.normal(shape=(256, 256))