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

Fix Metal API validation for empty concat (#1183)

Browse Source
This commit is contained in:
Awni Hannun
2024-06-04 13:17:08 -07:00
committed by GitHub
parent 375a8bbdcc
commit 83b11bc58d
2 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/tests/test_ops.py
View File

@@ -1533,6 +1533,12 @@ class TestOps(mlx_tests.MLXTestCase):
b = mx.array([1, 2])
mx.concatenate([a, b], axis=0)
# Cocnatenate with 0-sized array
a = mx.zeros((2, 0, 2))
b = mx.zeros((2, 2, 2))
out = mx.concatenate([a, b], axis=1)
self.assertTrue(mx.array_equal(out, b))
def test_meshgrid(self):
x = mx.array([1, 2, 3], dtype=mx.int32)
y = np.array([1, 2, 3], dtype=np.int32)