torch.cuda.nvtx.mark

torch.cuda.nvtx.mark(msg)[source][source]

描述在某个时刻发生的瞬时事件

参数:

msg (str) – 与事件关联的 ASCII 消息。