mindspore.runtime.current_stream

查看源文件
mindspore.runtime.current_stream()[源代码]

返回此设备上正在使用的流。

返回:

Stream,此设备上正在使用的流。

样例:

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