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

2025-10-19 16:48:10 +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/index.rst
+++ b/docs/src/index.rst
@@ -70,6 +70,7 @@ are the CPU and GPU.
   python/fft
   python/linalg
   python/metal
+   python/memory
   python/nn
   python/optimizers
   python/distributed
--- 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