zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-28 07:58:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix test tol

Browse Source
This commit is contained in:
Awni Hannun 2025-08-21 14:45:35 -07:00
parent 3ce23755ea
commit 80868ee4fb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/tests/test_quantized.py
View File

@ -341,7 +341,7 @@ class TestQuantized(mlx_tests.MLXTestCase):
) )
y_hat = x @ w_hat y_hat = x @ w_hat
self.assertEqual(y_q.shape, y_hat.shape) 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): def test_mode_error_cases(self):
w = mx.random.normal(shape=(256, 256)) w = mx.random.normal(shape=(256, 256))