torch.mtia.set_stream

torch.mtia.set_stream(stream)

设置当前流。这是一个设置流的包装器 API。

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

参数

stream (Stream) – 选定的流。如果此参数为 None，则此函数为无操作。