torch.mtia.set_stream

torch.mtia.set_stream(stream)

設定當前流。這是一個用於設定流的封裝 API。

不鼓勵使用此函式，推薦使用 stream 上下文管理器。

引數

stream (Stream) – 選定的流。如果此引數為 None，則此函式無效（no-op）。