TensorPlay
API symbolscuda
Copy
View MarkdownDownload .md

tensorplay.cuda.nvtx.range_end

tensorplay.cuda.nvtx.range_end(range_id) → None[source]

Mark the end of a range for a given range_id.

Parameters:

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

tensorplay.cuda.nvtx.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().

tensorplay.cuda.nvtx.range_pop

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

Ask DeepWiki