current_stream¶

paddle.device.cuda. current_stream ( device=None ) [source]

Return the current CUDA stream by the device.

Parameters

device (paddle.CUDAPlace()|int, optional) – The device or the ID of the device which want to get stream from. If device is None, the device is the current device. Default: None.

Returns

the stream to the device.

Return type

CUDAStream

Examples

>>> 
>>> import paddle
>>> paddle.device.set_device('gpu')

>>> s1 = paddle.device.cuda.current_stream()

>>> s2 = paddle.device.cuda.current_stream(0)

>>> s3 = paddle.device.cuda.current_stream(paddle.CUDAPlace(0))