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