nvtx_markers_ranges_nsight_profiling_quickstart.py

1import nvtx
2import time
3
4# Use a decorator to instrument a function
5@nvtx.annotate("my_function", color="blue")
6def my_function():
7    time.sleep(0.1)
8
9# Use a context manager to instrument a block of code
10with nvtx.annotate("my_block", color="green"):
11    time.sleep(0.1)
12
13# Use the low-level API to manually mark events
14nvtx.mark(message="event completed", color="red")
15
16my_function()