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

higher tol

Browse Source
This commit is contained in:
Awni Hannun 2025-04-29 12:41:07 -07:00
parent c384e4cbc6
commit 89d91a7c00
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

@ -572,7 +572,7 @@ class TestQuantized(mlx_tests.MLXTestCase):
out_down = mm(params, x, wq) out_down = mm(params, x, wq)
params[p][idx] += eps params[p][idx] += eps
num_ds = (out_up - out_down) / (2 * eps) num_ds = (out_up - out_down) / (2 * eps)
self.assertAlmostEqual(dparams[p][idx], num_ds, delta=1e-2) self.assertAlmostEqual(dparams[p][idx], num_ds, delta=2e-2)
if __name__ == "__main__": if __name__ == "__main__":