o i @s>ddlZddlZddlmZejdddZGdddZdS) N) time_rangeFccsFtdt|||| dVWddS1swYdS)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_colorsyncr R/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/cupy/prof/_time_range.pyrs"rc@s<eZdZdZejZejZejZej Z ej Z  dddZdS)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. NFcCstd|||||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)selfrrrr r r r __init__FszTimeRangeDecorator.__init__)NNNF) __name__ __module__ __qualname____doc__rrr __enter____exit____call__ _recreate_cmr r r r r 'sr )NNF) contextlibrcupyx.profilerrrcontextmanagerr r r r r s