tensorplay.cuda.nvtx API

Describe an instantaneous event that occurred at some point.

Parameters:

msg (str) – ASCII message to associate with the event.

Mark the end of a range for a given range_id.

Parameters:

range_id (int) – a unique handle for the start range.

Pop a range off of a stack of nested range spans. Returns the zero-based depth of the range that is ended.

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

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.

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