mirror of
https://github.com/ml-explore/mlx.git
synced 2025-07-18 23:21:16 +08:00
0eb56d5be0
* expose residency sets as wire/unwire * returns wired size * fix * runtime support check * fix os check * fix test * fix no metal build * docs * nit * nits in docs * nits
21 lines
297 B
ReStructuredText
21 lines
297 B
ReStructuredText
Metal
|
|
=====
|
|
|
|
.. currentmodule:: mlx.core.metal
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
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
|