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

fix pinv (#2110)

Browse Source
This commit is contained in:
Awni Hannun
2025-04-23 13:08:28 -07:00
committed by GitHub
parent 38c1e720c2
commit fbc89e3ced
7 changed files with 34 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/tests/test_array.py
View File

@@ -103,10 +103,12 @@ class TestDtypes(mlx_tests.MLXTestCase):
self.assertEqual(mx.finfo(mx.float32).min, np.finfo(np.float32).min)
self.assertEqual(mx.finfo(mx.float32).max, np.finfo(np.float32).max)
self.assertEqual(mx.finfo(mx.float32).eps, np.finfo(np.float32).eps)
self.assertEqual(mx.finfo(mx.float32).dtype, mx.float32)
self.assertEqual(mx.finfo(mx.float16).min, np.finfo(np.float16).min)
self.assertEqual(mx.finfo(mx.float16).max, np.finfo(np.float16).max)
self.assertEqual(mx.finfo(mx.float16).eps, np.finfo(np.float16).eps)
self.assertEqual(mx.finfo(mx.float16).dtype, mx.float16)
def test_iinfo(self):

5
python/tests/test_linalg.py
View File

@@ -232,6 +232,11 @@ class TestLinalg(mlx_tests.MLXTestCase):
for M, M_plus in zip(AB, pinvs):
self.assertTrue(mx.allclose(M @ M_plus @ M, M, rtol=0, atol=1e-3))
# Test singular matrix
A = mx.array([[4.0, 1.0], [4.0, 1.0]])
A_plus = mx.linalg.pinv(A, stream=mx.cpu)
self.assertTrue(mx.allclose(A @ A_plus @ A, A))
def test_cholesky_inv(self):
mx.random.seed(7)