torch.cuda.nvtx.range_push

torch.cuda.nvtx.range_push(msg)[source][source]

将一个范围推入嵌套范围栈。返回开始的范围的零基深度。

参数:

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