torch.accelerator.set_stream#
- torch.accelerator.set_stream(stream)[source]#
将当前流设置为给定的流。
- 参数
stream (torch.Stream) – 给定的流,该流必须与当前加速器设备类型匹配。
注意
此函数会将当前设备索引设置为给定流的设备索引。
将当前流设置为给定的流。
stream (torch.Stream) – 给定的流,该流必须与当前加速器设备类型匹配。
注意
此函数会将当前设备索引设置为给定流的设备索引。