set_stream

paddle.cuda. set_stream ( stream: Stream ) → Stream [source]

Set the current stream.

Parameters

stream (Stream) – The selected stream.

Returns

The previous stream.

Return type

Stream

Examples

>>> 
>>> 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)