mindspore.hal.set_cur_stream

mindspore.hal.set_cur_stream(stream)

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

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

参数：

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

异常：

• TypeError - 参数 stream 即不是一个 mindspore.hal.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())