NVIDIA / NVTX

The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resources in your applications.
Apache License 2.0
310 stars 48 forks source link

Python: add support for int and float payloads #90

Closed jfaust closed 9 months ago

jfaust commented 9 months ago

As referenced in #89

In addition to the basic unit tests, I've confirmed that nsys and Nsight systems see the payloads properly via a simple script:

import nvtx
import time

for i in range(0, 100):
    with nvtx.annotate('loop', payload=i):
        nvtx.mark('intvalue', payload=i)
        nvtx.mark('floatvalue', payload=float(i))
        time.sleep(0.01)

image image image