bindings for memory info (#761)

* bindings for memory info * update api * keep cache low if requested * fix default * nit in ops error
2025-09-01 04:24:36 +08:00 · 2024-03-01 19:51:58 -08:00
parent cf3eb87e52
commit d5964a2710
11 changed files with 300 additions and 69 deletions
--- a/docs/src/index.rst
+++ b/docs/src/index.rst
@@ -64,6 +64,7 @@ are the CPU and GPU.
   python/transforms
   python/fft
   python/linalg
+   python/metal
   python/nn
   python/optimizers
   python/tree_utils
--- a/docs/src/python/metal.rst
+++ b/docs/src/python/metal.rst
@@ -0,0 +1,14 @@
+Metal
+=====
+
+.. currentmodule:: mlx.core.metal
+
+.. autosummary:: 
+  :toctree: _autosummary
+
+  is_available
+  get_active_memory
+  get_peak_memory
+  get_cache_memory
+  set_memory_limit
+  set_cache_limit
--- a/docs/src/python/ops.rst
+++ b/docs/src/python/ops.rst
@@ -57,6 +57,7 @@ Operations
   greater_equal
   identity
   inner
+   isclose
   isnan
   isposinf
   isneginf
@@ -121,6 +122,7 @@ Operations
   tan
   tanh
   tensordot
+   tile
   transpose
   tri
   tril