Functions 6
mark
functionFull reference ↗range_end
functionFull reference ↗range_pop
functionFull reference ↗- tensorplay.cuda.nvtx.range_pop()[source]
Pop a range off of a stack of nested range spans. Returns the zero-based depth of the range that is ended.
range_push
functionFull reference ↗range_start
functionFull reference ↗- tensorplay.cuda.nvtx.range_start(msg) int[source]
Mark the start of a range with string message. It returns a unique handle for this range to pass to the corresponding call to rangeEnd().
Returns: A range handle (uint64_t) that can be passed to range_end().
- Parameters:
msg (str) – ASCII message to associate with the range.
range
functionFull reference ↗- tensorplay.cuda.nvtx.range(msg, *args, **kwargs)[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

