torch.cuda.nvtx.mark

torch.cuda.nvtx.mark(msg)

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

参数

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