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

Add tile op (#438)

Browse Source
This commit is contained in:
Diogo
2024-01-13 02:03:16 -05:00
committed by GitHub
parent 1b71487e1f
commit 2e29d0815b
7 changed files with 105 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
python/tests/test_ops.py
View File

@@ -1634,6 +1634,23 @@ class TestOps(mlx_tests.MLXTestCase):
np.allclose(np_out[0], mx_out[0]), msg=f"Shapes {s1} {s2}, Type {t}"
)
def test_tile(self):
self.assertCmpNumpy([(2,), [2]], mx.tile, np.tile)
self.assertCmpNumpy([(2, 3, 4), [2]], mx.tile, np.tile)
self.assertCmpNumpy([(2, 3, 4), [2, 1]], mx.tile, np.tile)
self.assertCmpNumpy(
[
(2, 3, 4),
[
2,
2,
],
],
mx.tile,
np.tile,
)
self.assertCmpNumpy([(3,), [2, 2, 2]], mx.tile, np.tile)
if __name__ == "__main__":
unittest.main()