mindspore.hal.set_cur_stream

mindspore.hal.set_cur_stream(stream)[源代码]

设置当前流,这是用于设置流的包装器API。

不建议使用此函数,建议使用 StreamCtx 上下文管理器。

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

异常:

样例:

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