change initial memory limits and add memory size to device info (#1064)

2025-11-04 10:38:10 +08:00 · 2024-05-03 06:50:15 -07:00
parent 02a9fc7bfa
commit b00ac960b4
3 changed files with 21 additions and 5 deletions
--- a/python/src/metal.cpp
+++ b/python/src/metal.cpp
@@ -129,6 +129,7 @@ void init_metal(nb::module_& m) {
      * ``architecture``
      * ``max_buffer_size``
      * ``max_recommended_working_set_size``
+      * ``memory_size``

      Returns:
          dict: A dictionary with string keys and string or integer values.