current_stream¶

paddle.device.cuda. current_stream ( device=None ) [源代码] ¶

通过 device 返回当前的 CUDA stream。

参数¶

device (paddle.CUDAPlace()|int，可选) - 希望获取 stream 的设备或者设备 ID。如果为 None，则为当前的设备。默认值为 None。

返回¶

CUDAStream，设备的 stream。

代码示例¶

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