set_stream

paddle.device. set_stream ( stream=None ) [源代码] 

设置当前的 stream。

参数

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

返回

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

代码示例

          >>> import paddle

>>> paddle.set_device('custom_cpu')
>>> s = paddle.device.Stream()
>>> # paddle.cuda.set_stream(s) is equivalent to paddle.device.set_stream(s)
>>> paddle.device.set_stream(s)

         