mlx.core.synchronize

目錄

mlx.core.synchronize#

synchronize(stream: Stream | mlx.core.ThreadLocalStream | Device | None = None) None#

Synchronize with the given stream.

參數:

stream (Stream, optional) -- Stream to synchronize. If device is provided the default stream for that device is used. If None then the default stream of the default device is used. Default: None.