mlx.core.get_array_buffer_size#
- get_array_buffer_size(*args) int#
Get the size of the buffers backing arrays in bytes.
Each unique buffer is counted once. The full allocator size of each buffer is used, which can exceed the logical size of its arrays. Arrays must be evaluated before calling this function. The result does not include graph objects, cached buffers, or other non-buffer memory.
- Parameters:
*args (arrays or trees of arrays) – Each argument can be a single array or a tree of arrays. Leaves which are not arrays are ignored.
- Returns:
The size of the unique buffers in bytes.
- Return type: