评价此页

torch.cuda.nvtx.range_push#

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

将范围推送到范围跨度的堆栈。返回已启动范围的零基深度。

参数

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