fix cpu compile (#1897)

2025-11-04 18:48:15 +08:00 · 2025-02-24 14:10:30 -08:00
parent a44dc4bdb0
commit 7face5d9fd
2 changed files with 6 additions and 2 deletions
--- a/python/tests/test_compile.py
+++ b/python/tests/test_compile.py
@@ -953,6 +953,7 @@ class TestCompile(mlx_tests.MLXTestCase):
        self.assertEqual(out[1].shape, (2, 2, 5))

    def test_leaks(self):
+        gc.collect()
        if mx.metal.is_available():
            mem_pre = mx.metal.get_active_memory()
        else: