mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-25 18:11:15 +08:00
3d405fb3b1
* 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
20 lines
278 B
ReStructuredText
20 lines
278 B
ReStructuredText
.. _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
|