Shortcuts

torch.cuda.nvtx.range_push

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

将一个范围推入嵌套范围跨度的堆栈。返回已开始范围的从零开始的深度。

Parameters

msg (str) – 与范围关联的ASCII消息