no segfault with uninitialized array.at (#2514)

This commit is contained in:
Awni Hannun
2025-08-18 08:33:38 -07:00
committed by GitHub
parent c5fcd5b61b
commit e7c6e1db82
2 changed files with 18 additions and 0 deletions

View File

@@ -1365,6 +1365,9 @@ class TestArray(mlx_tests.MLXTestCase):
def test_array_at(self):
a = mx.array(1)
with self.assertRaises(ValueError):
a.at.add(1)
a = a.at[None].add(1)
self.assertEqual(a.item(), 2)