评价此页

torch.cuda.nvtx.range_push#

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

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

参数

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