mindspore.runtime.set_cur_stream

mindspore.runtime.set_cur_stream(stream)

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

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

参数：

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

异常：

• TypeError - 参数 stream 既不是 mindspore.runtime.Stream 也不是 None。

支持平台：

Ascend GPU

样例：

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