mindspore.runtime.current_stream

mindspore.runtime.current_stream()

Return current stream used on this device.

Returns

stream (Stream), current stream.

Supported Platforms:

Ascend GPU

Examples

>>> import mindspore as ms
>>> ms.set_device("Ascend", 0)
>>> cur_stream = ms.runtime.current_stream()
>>> assert cur_stream == ms.runtime.default_stream()