move memory APIs into top level mlx.core (#1982)

2025-12-16 01:49:05 +08:00 · 2025-03-21 07:25:12 -07:00
parent 65a38c452b
commit 4e1994e9d7
25 changed files with 418 additions and 323 deletions
--- a/docs/src/python/memory.rst
+++ b/docs/src/python/memory.rst
@@ -0,0 +1,16 @@
+Memory Management
+=================
+
+.. currentmodule:: mlx.core
+
+.. autosummary::
+  :toctree: _autosummary
+
+  get_active_memory
+  get_peak_memory
+  reset_peak_memory
+  get_cache_memory
+  set_memory_limit
+  set_cache_limit
+  set_wired_limit
+  clear_cache