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

Increase test tolerance for fast.layer_norm (#880)

Browse Source
This commit is contained in:
Angelos Katharopoulos 2024-03-22 12:10:27 -07:00 committed by GitHub
parent 9663c22fe9
commit fcda3a0e66
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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_fast.py
View File

@ -181,7 +181,7 @@ class TestFast(mlx_tests.MLXTestCase):
return x return x
# Per dtype absolute tolerance # Per dtype absolute tolerance
tolerances = {mx.float32: 2e-6, mx.float16: 2e-3, mx.bfloat16: 2e-2} tolerances = {mx.float32: 3e-6, mx.float16: 3e-3, mx.bfloat16: 3e-2}
dtypes = [mx.float32, mx.float16, mx.bfloat16] dtypes = [mx.float32, mx.float16, mx.bfloat16]
epss = [1e-3, 1e-5] epss = [1e-3, 1e-5]