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

Bump nanobind to 2.2 (#1461)

Browse Source 
* bump nanobind

* extension version for tests
This commit is contained in:
Awni Hannun
2024-10-07 16:52:40 -07:00
committed by GitHub
parent 0070e1db40
commit f374b6ca4d
6 changed files with 17 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
python/tests/test_array.py
View File

@@ -688,6 +688,13 @@ class TestArray(mlx_tests.MLXTestCase):
self.assertEqual(b_npy.dtype, np_dtype)
def test_array_from_noncontiguous_np(self):
for t in [np.int8, np.int32, np.float16, np.float32, np.complex64]:
np_arr = np.random.uniform(size=(10, 10)).astype(np.complex64)
np_arr = np_arr.T
mx_arr = mx.array(np_arr)
self.assertTrue(mx.array_equal(np_arr, mx_arr))
def test_array_np_shape_dim_check(self):
a_npy = np.empty(2**31, dtype=np.bool_)
with self.assertRaises(ValueError) as e: