`i @ddlZddlmZddlmZGddZdS)N)cuda)runtimec4eZdZdZ d dZdZdZdZdZdS) time_rangeaMark function calls with ranges using NVTX/rocTX. This object can be used either as a decorator or a context manager. When used as a decorator, the decorated function calls are marked as ranges: >>> from cupyx.profiler import time_range >>> @time_range() ... def function_to_profile(): ... pass When used as a context manager, it describes the enclosed block as a nested range: >>> from cupyx.profiler import time_range >>> with time_range('some range in green', color_id=0): ... # do something you want to measure ... pass The marked ranges are visible in the profiler (such as nvvp, nsys-ui, etc) timeline. Args: message (str): Name of a range. When used as a decorator, the default is ``func.__name__``. color_id: range color ID argb_color: range color in ARGB (e.g. 0xFF00FF00 for green) sync (bool): If ``True``, waits for completion of all outstanding processing on GPU before calling :func:`cupy.cuda.nvtx.RangePush()` or :func:`cupy.cuda.nvtx.RangePop()` .. seealso:: :func:`cupy.cuda.nvtx.RangePush`, :func:`cupy.cuda.nvtx.RangePop` NFctjjstd||t d||_||nd|_||_||_dS)Nznvtx is not installedz3Only either color_id or argb_color can be specified) rnvtx available RuntimeError ValueErrormessagecolor_id argb_colorsync)selfr rrrs n/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/profiler/_time_range.py__init__ztime_range.__init__/siy" 8677 7  J$:E  $,$8b $ c|jtd|jrtj|j+t j|j|jn*t j |j|j |S)NzCwhen used as a context manager, the message argument cannot be None) r r rrdeviceSynchronizerrr RangePushC RangePushr)rs r __enter__ztime_range.__enter__=s <   9 (  % ' ' ' ? & I t ? ? ? ? I   dm < < < rcv|jrtjtjdSN)rrrrr RangePop)rexc_type exc_value tracebacks r__exit__ztime_range.__exit__Js5 9 (  % ' ' ' rc"|j||_|Sr)r )rr s r _recreate_cmztime_range._recreate_cmOs < "DL rcJtjfd}|S)Nc|j5|i|cdddS#1swxYwYdSr)r"__name__)argskwargsfuncrs rinnerz"time_range.__call__..innerUs""4=11 - -tT,V,, - - - - - - - - - - - - - - - - - -s 155) functoolswraps)rr(r)s`` r__call__ztime_range.__call__Ts>    - - - - -   - r)NNNF) r% __module__ __qualname____doc__rrr r"r,rrrr sv!!HFK         rr)r*cupyrcupy_backends.cuda.apirrr0rrr3sq******NNNNNNNNNNr