cupy.cuda.get_current_stream

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

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

参数:

device_id (int, optional) – 要检查当前流的设备索引。默认情况下选择当前活动的设备。

返回:

当前的 CUDA 流。

返回类型:

cupy.cuda.Stream