torch.cuda.nvtx.range¶
- torch.cuda.nvtx.range(msg, *args, **kwargs)[source][source]¶
Context manager / decorator that pushes an NVTX range at the beginning of its scope, and pops it at the end. If extra arguments are given, they are passed as arguments to msg.format().
- Parameters
msg (str) – message to associate with the range
