`i ^ddlZddlZddlmZejddZGddZdS)N) time_rangeFc#Ktjdt||||5dVddddS#1swxYwYdS)aA context manager to describe the enclosed block as a nested range >>> from cupy import prof >>> with cupy.prof.time_range('some range in green', color_id=0): ... # do something you want to measure ... pass Args: message: Name of a range. 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` .. warning:: This context manager is deprecated. Please use :class:`cupyx.profiler.time_range` instead. zcupy.prof.time_range has been deprecated since CuPy v10 and will be removed in the future. Use cupyx.profiler.time_range instead.N)warningswarn _time_range)messagecolor_id argb_colorsyncs i/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/prof/_time_range.pyrrs. M  Wh D 9 9 s ;??cdeZdZdZejZejZejZej Z ej Z ddZdS)TimeRangeDecoratoraZDecorator to mark function calls with range in NVIDIA profiler Decorated function calls are marked as ranges in NVIDIA profiler timeline. >>> from cupy import prof >>> @cupy.prof.TimeRangeDecorator() ... def function_to_profile(): ... pass Args: message (str): Name of a range, default use ``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` .. warning:: This decorator is deprecated. Please use :class:`cupyx.profiler.time_range` instead. NFc^tjd|||||dS)Nzcupy.prof.TimeRangeDecorator has been deprecated since CuPy v10 and will be removed in the future. Use cupyx.profiler.time_range instead.)rr_init)selfrr r r s r __init__zTimeRangeDecorator.__init__Fs=       7Hj$77777)NNNF) __name__ __module__ __qualname____doc__rrr __enter____exit____call__ _recreate_cmrr rr'sb0  E%I#H#H+L?C888888rr)NNF) contextlibrcupyx.profilerrrcontextmanagerrrrr r s444444 >'8'8'8'8'8'8'8'8'8'8r