set_stream¶

参数¶

• stream (paddle.device.Stream) - 希望设置的 stream。

返回¶

paddle.device.Stream，设备之前的 stream。

代码示例¶

# required: custom_device
import paddle

paddle.set_device('custom_cpu')
s = paddle.device.Stream()
paddle.device.set_stream(s)