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

This commit is contained in:
Awni Hannun
2024-05-03 06:50:15 -07:00
committed by GitHub
parent 02a9fc7bfa
commit b00ac960b4
3 changed files with 21 additions and 5 deletions

View File

@@ -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.