mirror of
https://github.com/ml-explore/mlx.git
synced 2025-10-19 00:04:41 +08:00
No copy command encoder (#986)
* no copy command encoder * up layer norm test tolerances
This commit is contained in:
@@ -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]
|
||||
|
Reference in New Issue
Block a user