torch.cuda.nvtx.range_push
torch.cuda.nvtx.range_push(msg)
Pushes 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