mirror of
https://github.com/ml-explore/mlx.git
synced 2025-10-19 00:04:41 +08:00
Fix copy bug with many dims (#1137)
This commit is contained in:
@@ -1270,6 +1270,10 @@ class TestArray(mlx_tests.MLXTestCase):
|
||||
x[:, 0] = 1.0
|
||||
self.assertTrue(mx.array_equal(x[:, 0], mx.ones((2, 4, 5, 3))))
|
||||
|
||||
x = mx.zeros((2, 2, 2, 2, 2, 2))
|
||||
x[0, 0] = 1
|
||||
self.assertTrue(mx.array_equal(x[0, 0], mx.ones((2, 2, 2, 2))))
|
||||
|
||||
def test_array_at(self):
|
||||
a = mx.array(1)
|
||||
a = a.at[None].add(1)
|
||||
|
Reference in New Issue
Block a user