torch.cuda.nvtx.range_push

torch.cuda.nvtx.range_push(msg)

Push a range onto a stack of nested range span. Returns zero-based depth of the range that is started.

Parameters

msg (str) – ASCII message to associate with range