Shortcuts

torch.cuda.nvtx.mark

torch.cuda.nvtx.mark(msg)[源代码]

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

Parameters

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