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

No copy command encoder (#986)

Browse Source 
* no copy command encoder

* up layer norm test tolerances
This commit is contained in:
Awni Hannun
2024-04-11 21:15:36 -07:00
committed by GitHub
parent 91eba8e485
commit ae18326533
3 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/tests/test_fast.py
View File

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