nvtx_python_profiling_with_decorators_context_managers_and_ranges.py

1import nvtx
2import time
3
4# 1. Using a decorator to instrument a function
5@nvtx.annotate("my_function", color="blue")
6def my_function():
7    time.sleep(0.1)
8
9# 2. Using a context manager to instrument a block of code
10with nvtx.annotate("my_block", color="green"):
11    time.sleep(0.1)
12
13# 3. Using manual push and pop for fine-grained control
14nvtx.push_range("manual_range", color="red")
15time.sleep(0.1)
16nvtx.pop_range()
17
18if __name__ == "__main__":
19    my_function()