torch.cuda.nvtx.range_push

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

将一个范围压入嵌套范围栈中，并返回该范围的零基础深度。

参数

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