Limit compile buffers (#1887)

* limit compile buffers * maybe not flaky test
2025-11-03 18:18:15 +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/test_load.py
+++ b/python/tests/test_load.py
@@ -385,6 +385,7 @@ class TestLoad(mlx_tests.MLXTestCase):
        mx.eval(x)
        save_file = os.path.join(self.test_dir, "donation.npy")
        mx.save(save_file, x)
+        mx.synchronize(mx.default_stream(mx.default_device()))

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