mlx.core.set_default_stream

set_default_stream(stream: Stream) → None

Set the default stream.

This will make the given stream the default for the streams device. It will not change the default device.

参数:

stream (stream) -- Stream to make the default.