mindspore.hal.set_cur_stream

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

设置当前流,此接口将在后续版本中废弃,请使用接口 mindspore.runtime.set_cur_stream() 代替。

参数:
  • stream (Stream) - 指定的流。如果是 None ,这个上下文管理器无操作。

样例:

>>> import mindspore
>>> cur_stream = mindspore.hal.current_stream()
>>> assert cur_stream == mindspore.hal.default_stream()
>>> s1 = mindspore.hal.Stream()
>>> mindspore.hal.set_cur_stream(s1)
>>> assert mindspore.hal.current_stream() == s1
>>> mindspore.hal.set_cur_stream(mindspore.hal.default_stream())