cupy.cuda.get_current_stream#

cupy.cuda.get_current_stream(int device_id=-1)#

Gets the current CUDA stream for the specified CUDA device.

Parameters:

device_id (int, optional) – Index of the device to check for the current stream. The currently active device is selected by default.

Returns:

The current CUDA stream.

Return type:

cupy.cuda.Stream