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())