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

some fixes (#1281)

Browse Source
This commit is contained in:
Awni Hannun
2024-07-23 11:49:05 -07:00
committed by GitHub
parent 6768c6a54a
commit e2aa6ec8ae
4 changed files with 14 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
python/tests/test_ops.py
View File

@@ -847,9 +847,10 @@ class TestOps(mlx_tests.MLXTestCase):
def test_expm1(self):
a = mx.array([-88, -87, 0, 0.5, -0.5, 5, 87, 88, 89, 90])
result = mx.expm1(a)
expected = np.expm1(a, dtype=np.float32)
self.assertTrue(np.allclose(result, expected, rtol=1e-5, atol=1e-5))
errs = np.seterr(over="ignore")
expected = np.expm1(a)
np.seterr(over=errs["over"])
self.assertTrue(np.allclose(result, expected, rtol=1e-3, atol=1e-4))
def test_erf(self):
inputs = [-5, 0.0, 0.5, 1.0, 2.0, 10.0]