mlx.core.stream

stream(s: Union[Stream, Device]) → mlx.core.StreamContext

Create a context manager to set the default device and stream.

Parameters:

s – The Stream or Device to set as the default.

Returns:

A context manager that sets the default device and stream.

Example: