rebase

2025-09-18 10:26:56 +08:00 · 2025-03-24 20:24:41 +00:00
parent 0438cc01cd
commit 450ede431c
3033 changed files with 9909 additions and 442035 deletions
--- a/docs/build/html/_sources/python/_autosummary/mlx.core.clear_cache.rst
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.clear_cache.rst
@@ -0,0 +1,6 @@
+mlx.core.clear\_cache
+=====================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: clear_cache
--- a/docs/build/html/_sources/python/_autosummary/mlx.core.get_active_memory.rst
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.get_active_memory.rst
@@ -0,0 +1,6 @@
+mlx.core.get\_active\_memory
+============================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: get_active_memory
--- a/docs/build/html/_sources/python/_autosummary/mlx.core.get_cache_memory.rst
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.get_cache_memory.rst
@@ -0,0 +1,6 @@
+mlx.core.get\_cache\_memory
+===========================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: get_cache_memory
--- a/docs/build/html/_sources/python/_autosummary/mlx.core.get_peak_memory.rst
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.get_peak_memory.rst
@@ -0,0 +1,6 @@
+mlx.core.get\_peak\_memory
+==========================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: get_peak_memory
--- a/docs/build/html/_sources/python/_autosummary/mlx.core.metal.clear_cache.rst
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.metal.clear_cache.rst
@@ -1,6 +0,0 @@
-mlx.core.metal.clear\_cache
-===========================
-
-.. currentmodule:: mlx.core.metal
-
-.. autofunction:: clear_cache
--- a/docs/build/html/_sources/python/_autosummary/mlx.core.metal.get_active_memory.rst
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.metal.get_active_memory.rst
@@ -1,6 +0,0 @@
-mlx.core.metal.get\_active\_memory
-==================================
-
-.. currentmodule:: mlx.core.metal
-
-.. autofunction:: get_active_memory
--- a/docs/build/html/_sources/python/_autosummary/mlx.core.metal.get_cache_memory.rst
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.metal.get_cache_memory.rst
@@ -1,6 +0,0 @@
-mlx.core.metal.get\_cache\_memory
-=================================
-
-.. currentmodule:: mlx.core.metal
-
-.. autofunction:: get_cache_memory
--- a/docs/build/html/_sources/python/_autosummary/mlx.core.metal.get_peak_memory.rst
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.metal.get_peak_memory.rst
@@ -1,6 +0,0 @@
-mlx.core.metal.get\_peak\_memory
-================================
-
-.. currentmodule:: mlx.core.metal
-
-.. autofunction:: get_peak_memory
--- a/docs/build/html/_sources/python/_autosummary/mlx.core.metal.reset_peak_memory.rst
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.metal.reset_peak_memory.rst
@@ -1,6 +0,0 @@
-mlx.core.metal.reset\_peak\_memory
-==================================
-
-.. currentmodule:: mlx.core.metal
-
-.. autofunction:: reset_peak_memory
--- a/docs/build/html/_sources/python/_autosummary/mlx.core.metal.set_cache_limit.rst
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.metal.set_cache_limit.rst
@@ -1,6 +0,0 @@
-mlx.core.metal.set\_cache\_limit
-================================
-
-.. currentmodule:: mlx.core.metal
-
-.. autofunction:: set_cache_limit
--- a/docs/build/html/_sources/python/_autosummary/mlx.core.metal.set_memory_limit.rst
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.metal.set_memory_limit.rst
@@ -1,6 +0,0 @@
-mlx.core.metal.set\_memory\_limit
-=================================
-
-.. currentmodule:: mlx.core.metal
-
-.. autofunction:: set_memory_limit
--- a/docs/build/html/_sources/python/_autosummary/mlx.core.metal.set_wired_limit.rst
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.metal.set_wired_limit.rst
@@ -1,6 +0,0 @@
-mlx.core.metal.set\_wired\_limit
-================================
-
-.. currentmodule:: mlx.core.metal
-
-.. autofunction:: set_wired_limit
--- a/docs/build/html/_sources/python/_autosummary/mlx.core.reset_peak_memory.rst
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.reset_peak_memory.rst
@@ -0,0 +1,6 @@
+mlx.core.reset\_peak\_memory
+============================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: reset_peak_memory
--- a/docs/build/html/_sources/python/_autosummary/mlx.core.set_cache_limit.rst
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.set_cache_limit.rst
@@ -0,0 +1,6 @@
+mlx.core.set\_cache\_limit
+==========================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: set_cache_limit
--- a/docs/build/html/_sources/python/_autosummary/mlx.core.set_memory_limit.rst
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.set_memory_limit.rst
@@ -0,0 +1,6 @@
+mlx.core.set\_memory\_limit
+===========================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: set_memory_limit
--- a/docs/build/html/_sources/python/_autosummary/mlx.core.set_wired_limit.rst
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.set_wired_limit.rst
@@ -0,0 +1,6 @@
+mlx.core.set\_wired\_limit
+==========================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: set_wired_limit
--- a/docs/build/html/_sources/python/memory_management.rst
+++ b/docs/build/html/_sources/python/memory_management.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
--- a/docs/build/html/_sources/python/metal.rst
+++ b/docs/build/html/_sources/python/metal.rst
@@ -8,13 +8,5 @@ Metal

  is_available
  device_info
-  get_active_memory
-  get_peak_memory
-  reset_peak_memory
-  get_cache_memory
-  set_memory_limit
-  set_cache_limit
-  set_wired_limit
-  clear_cache
  start_capture
  stop_capture