mlx/docs/src/python/metal.rst

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
  clear_cache
  start_capture
  stop_capture
bindings for memory info (#761) * bindings for memory info * update api * keep cache low if requested * fix default * nit in ops error 2024-03-02 11:51:58 +08:00			`Metal`
			`=====`

			`.. currentmodule:: mlx.core.metal`

Improve profiling with gpu tracing (#969) * improve profiling with gpu tracing * fix for linux * nit * doc fix * fix example 2024-04-08 12:47:43 +08:00			`.. autosummary::`
bindings for memory info (#761) * bindings for memory info * update api * keep cache low if requested * fix default * nit in ops error 2024-03-02 11:51:58 +08:00			`:toctree: _autosummary`

			`is_available`
Add a `mx.metal.device_info` (#1060) * device inof * add variant * fix linux * fix doc 2024-05-01 06:47:27 +08:00			`device_info`
bindings for memory info (#761) * bindings for memory info * update api * keep cache low if requested * fix default * nit in ops error 2024-03-02 11:51:58 +08:00			`get_active_memory`
			`get_peak_memory`
Reset peak memory (#1074) * reset peak memory * fix linux * nits in docs 2024-05-04 08:12:51 +08:00			`reset_peak_memory`
bindings for memory info (#761) * bindings for memory info * update api * keep cache low if requested * fix default * nit in ops error 2024-03-02 11:51:58 +08:00			`get_cache_memory`
			`set_memory_limit`
			`set_cache_limit`
Expose function to clear memory cache (#1032) * expose function to clear memory cache * fix linux build * fix metal tests 2024-04-25 07:48:51 +08:00			`clear_cache`
Improve profiling with gpu tracing (#969) * improve profiling with gpu tracing * fix for linux * nit * doc fix * fix example 2024-04-08 12:47:43 +08:00			`start_capture`
			`stop_capture`