torch.cuda.nvtx.mark

torch.cuda.nvtx.mark(msg)

描述在某个时间点发生的瞬时事件。

参数

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