mlx.core.synchronize

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

Synchronize with the given stream.

Parameters:

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.