nvtx_python_profiling_markers_ranges_nsight_quickstart.py

1import nvtx
2import time
3
4# Annotate a function as an NVTX range
5@nvtx.annotate("my_function", color="blue")
6def my_function():
7    # Use a manual range
8    with nvtx.annotate("inner_range", color="green"):
9        time.sleep(0.1)
10    
11    # Mark a discrete event in time
12    nvtx.mark(message="event_happened", color="red")
13    time.sleep(0.1)
14
15if __name__ == "__main__":
16    my_function()