Limit compile buffers (#1887)

* limit compile buffers * maybe not flaky test
2025-11-01 16:58:08 +08:00 · 2025-02-19 20:28:13 -08:00
parent 78ba24c37d
commit c707b2b0a6
4 changed files with 43 additions and 1 deletions
--- a/python/tests/mpi_test_distributed.py
+++ b/python/tests/mpi_test_distributed.py
@@ -177,6 +177,7 @@ class TestDistributed(mlx_tests.MLXTestCase):
    def test_donation(self):
        x = mx.random.normal((1024,))
        mx.eval(x)
+        mx.synchronize(mx.default_stream(mx.default_device()))

        mx.metal.reset_peak_memory()
        scale = mx.array(2.0)