cuda fix flaky test (#2646)

2025-10-19 08:20:15 +08:00 · 2025-10-02 15:40:04 -07:00
parent c2c3e0b0a2
commit 1c9ae1eaa1
2 changed files with 2 additions and 0 deletions
--- a/python/tests/test_array.py
+++ b/python/tests/test_array.py
@@ -2005,6 +2005,7 @@ class TestArray(mlx_tests.MLXTestCase):
            b = mx.reshape(b, [])
            return b

+        mx.synchronize()
        t()
        gc.collect()
        expected = get_mem()