PyNVTX 0.1.2

A thin python wrapper for the nvToolsExt (NVTX) library, using pybind11

PyNVTX

A thin python wrapper for the nvToolsExt (NVTX) library, using pybind11. This wrapper is meant to be as thin as possible -- so only provides minimal support. Currently supported features are:

1. NVTX markers: nvtxRangePushA and nvtxRangePop
2. Function decorator: PyNVTX.wrap

NVTX Markers (nvtxRangePushA / nvtxRangePop)

import PyNVTX as nvtx

nvtx.RangePushA("Doing some work")

# code to time goes here

nvtx.RangePop()

Function Decorator

This will put RangePushA and RangePop the the beginning and and of the function call:

@nvtx.mark("test_function")
def test():
    # You code goes here

Example Code

To get you started, take a look at test/test-nvtx.py