torch.cuda.nvtx.mark

torch.cuda.nvtx.mark(msg)

描述在某個時間點發生的瞬時事件。

引數

msg (str) – 要與事件關聯的 ASCII 訊息。