9j DSSKJr SSKrSSKJr SSKJr "SS5rg)) annotationsN)cuda)runtimec<\rSrSrSrS SjrSrSrSrSr S r g) time_range aOMark 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` Nc[RR(d [S5eUbUb [ S5eXlUbUOSUlX0lX@lg)Nznvtx is not installedz3Only either color_id or argb_color can be specified) rnvtx available RuntimeError ValueErrormessagecolor_id argb_colorsync)selfrrrrs T/home/wildlama/miniconda3/lib/python3.13/site-packages/cupyx/profiler/_time_range.py__init__time_range.__init__1sUyy""67 7  J$:E  $,$8b $ cpURc [S5eUR(a[R"5 UR b6[ RRURUR 5 U$[ RRURUR5 U$)NzCwhen used as a context manager, the message argument cannot be None) rrrrdeviceSynchronizerrr RangePushC RangePushr)rs r __enter__time_range.__enter__?s <<   99  % % ' ?? & II t ?  II   dmm < rcUR(a[R"5 [RR 5 gN)rrrrr RangePop)rexc_type exc_value tracebacks r__exit__time_range.__exit__Ls% 99  % % ' rc,URcXlU$r)r)rrs r _recreate_cmtime_range._recreate_cmQs << "L rcJ^^[R"T5UU4Sj5nU$)Nc>TRTR5 T"U0UD6sSSS5 $!,(df  g=fr)r'__name__)argskwargsfuncrs rinner"time_range.__call__..innerWs.""4==1T,V,211s/ =) functoolswraps)rr.r/s`` r__call__time_range.__call__Vs%   -  - r)rrrr)NNNF) r+ __module__ __qualname____firstlineno____doc__rrr$r'r3__static_attributes__rrrr s(!HFK    rr) __future__rr1cupyrcupy_backends.cuda.apirrr:rrr>s"*NNr