torch.accelerator.set_stream

torch.accelerator.set_stream(stream)

將目前的 stream 設定為指定的 stream。

參數

stream (torch.Stream) – 指定的 stream，必須符合目前的 加速器裝置類型。

注意

此函數會將目前的裝置索引設定為指定 stream 的裝置索引。