o Xغ·iشم@s>ddlmZddlZddlmZddlmZGdd„dƒZdS)é)عannotationsN)عcuda)عruntimec@s<eZdZdZ ddd„Zdd„Zdd „Zd d„Zdd „ZdS)ع time_rangea·Mark 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` NFcCsPtjjstdƒ‚|dur|durtdƒ‚||_|dur|nd|_||_||_dS)Nznvtx is not installedz3Only either color_id or argb_color can be specifiedéےےےے) rعnvtxع availableعRuntimeErrorع ValueErrorعmessageعcolor_idع argb_colorعsync)عselfrrr r©rْP/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupyx/profiler/_time_range.pyع__init__1sے ztime_range.__init__cCsV|jdur tdƒ‚|jrt ،|jdur tj |j|j،|Stj |j|j ،|S)NzCwhen used as a context manager, the message argument cannot be None)rr rrعdeviceSynchronizer rrع RangePushCع RangePushr)rrrrع __enter__?s ے ےztime_range.__enter__cCs|jrt ،tj ،dS©N)rrrrrعRangePop)rعexc_typeع exc_valueع tracebackrrrع__exit__Lsztime_range.__exit__cCs|jdur||_|Sr)r)rrrrrع_recreate_cmQs ztime_range._recreate_cmcst ˆ،‡‡fdd„ƒ}|S)Ncs>ˆ ˆj،ڈˆ|i|¤ژWdƒS1swYdSr)rع__name__)عargsعkwargs©عfuncrrrعinnerWs$ےz"time_range.__call__..inner)ع functoolsعwraps)rr"r#rr!rع__call__Vsztime_range.__call__)NNNF) rع __module__ع__qualname__ع__doc__rrrrr&rrrrr s$ ے r)ع __future__rr$عcupyrعcupy_backends.cuda.apirrrrrrعs