torch.cuda.set_stream

torch.cuda.set_stream(stream)

Set the current stream.This is a wrapper API to set the stream.

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

Parameters

流 (流) – 选定的流。如果此参数为 None，则此函数不执行任何操作。