Fix slice data size (#1913)

* fix slice data size * add test
2025-09-01 12:49:44 +08:00 · 2025-03-02 21:50:42 -08:00
parent 5e6c130d93
commit 4e7cd31d12
2 changed files with 13 additions and 3 deletions
--- a/python/tests/test_ops.py
+++ b/python/tests/test_ops.py
@@ -2846,6 +2846,11 @@ class TestOps(mlx_tests.MLXTestCase):
        b[::2] = 0
        self.assertTrue(mx.array_equal(b, mx.array([0, 3, 0, 1])))

+    def test_slice_with_negative_stride(self):
+        a = mx.random.uniform(shape=(128, 4))
+        out = a[::-1]
+        self.assertTrue(mx.array_equal(out[-1, :], a[0, :]))
+

 if __name__ == "__main__":
    unittest.main()