cupy.cuda.nvtx.Mark

cupy.cuda.nvtx.Mark(message, int id_color=-1)

标记应用程序中的瞬时事件（标记）。

标记用于描述应用程序执行期间特定时间点的事件。

参数:

• message (str) – 标记的名称。

• id_color (int) – 标记的颜色 ID。