torch.mtia.set_stream

torch.mtia.set_stream(stream)

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

Usage of this function is discouraged in favor of the stream context manager.

引數

stream (Stream) – 選擇的流。如果此引數為 None，則此函式無效。