torch.cuda.nvtx.range

torch.cuda.nvtx.range(msg, *args, **kwargs)[源代码]

这是一个上下文管理器和装饰器，它在作用域开始时推送一个NVTX范围，并在作用域结束时弹出该范围。如果有额外的参数，这些参数会被传递给msg.format()方法。