mindspore.runtime.default_stream

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

返回此设备上的默认流。

返回:

Stream,此设备上的默认流。

样例:

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