mindspore.hal.current_stream

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

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

返回:

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

样例:

>>> import mindspore as ms
>>> cur_stream = ms.hal.current_stream()
>>> assert cur_stream == ms.hal.default_stream()