cupy.cuda.get_current_stream

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

获取指定 CUDA 设备的当前 CUDA 流。

参数:

device_id (int, 可选) – 检查当前流的设备索引。默认选择当前活动设备。

返回:

当前 CUDA 流。

返回类型:

cupy.cuda.Stream