mlx/docs/src/python/devices_and_streams.rst

.. _devices_and_streams:

Devices and Streams
===================

.. currentmodule:: mlx.core

.. autosummary::
  :toctree: _autosummary

   Device
   Stream
   default_device
   set_default_device
   default_stream
   new_stream
   set_default_stream
   stream
   synchronize
jagrit's commit files 2023-11-30 02:52:08 +08:00			`.. _devices_and_streams:`

			`Devices and Streams`
			`===================`

			`.. currentmodule:: mlx.core`

			`.. autosummary::`
			`:toctree: _autosummary`

			`Device`
Adds device context manager (#679) 2024-02-15 06:14:58 +08:00			`Stream`
jagrit's commit files 2023-11-30 02:52:08 +08:00			`default_device`
			`set_default_device`
			`default_stream`
			`new_stream`
			`set_default_stream`
Adds device context manager (#679) 2024-02-15 06:14:58 +08:00			`stream`
Add synchronize function (#1006) * add synchronize function * fix linux * fix linux * fix and fix docs * fix test * try synchronize in stream destroy * synchronize works for both cpu and gpu 2024-04-22 23:25:46 +08:00			`synchronize`