current_stream

paddle.cuda. current_stream ( device: DeviceLike = None ) → Stream [source]

Return the current stream for the given device.

Parameters

device (int | str | paddle.CUDAPlace | paddle.CustomPlace | None, optional) –

The target device to query.

• None: use the current device.

• int: device index (e.g., 0 -> ‘gpu:0’).

• str: device string (e.g., “cuda:0”, “gpu:1”).

• CUDAPlace or CustomPlace: Paddle device objects.

Returns

The current CUDA stream associated with the given device.

Return type

core.CUDAStream

Examples

>>> 
>>> import paddle

# Get the current stream on the default CUDA device
>>> s1 = paddle.cuda.current_stream()
>>> print(s1)

# Get the current stream on device cuda:0
>>> s2 = paddle.cuda.current_stream("cuda:0")
>>> print(s2)