From fcda3a0e6621425c7b4d0a6a7e402fed9712dcee Mon Sep 17 00:00:00 2001 From: Angelos Katharopoulos Date: Fri, 22 Mar 2024 12:10:27 -0700 Subject: [PATCH] Increase test tolerance for fast.layer_norm (#880) --- python/tests/test_fast.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tests/test_fast.py b/python/tests/test_fast.py index 77b5c721f..a2b803d70 100644 --- a/python/tests/test_fast.py +++ b/python/tests/test_fast.py @@ -181,7 +181,7 @@ class TestFast(mlx_tests.MLXTestCase): return x # 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] epss = [1e-3, 1e-5]