torch.cuda.nvtx.mark ==================== .. currentmodule:: torch.cuda.nvtx .. autofunction:: mark