Foj <SSKrSSKrSSKJr SSKJrJrJrJr SSK J r SSK J r J r SSK Jr \R "5r\\\\\\\4r"SS 5r"S S \5rSS .S \S\\\\4S\\4SjjrSSS .S \\S\\\\4S\\4Sjjjrg)Nwraps)OptionalUnionAnySequence)flags)transform_tensor_metricsset_metric_kernels)protonc`\rSrSrSrSS\S\\\\4SS4Sjjr Sr S r S r S r S rS rg)scopea A context manager and decorator for entering and exiting a scope. Usage: context manager: ```python with proton.scope("test0", {metric_name: metric_value}): foo[1,](x, y) ``` decorator: ```python @proton.scope("test0", {metric_name: metric_value}) def foo(x, y): ... ``` Args: name (str): The name of the scope. metrics (dict[str, float], optional): The metrics of the scope. Default is None. Nnamemetricsreturnc*XlX lSUlgNrrid)selfrrs O/mnt/workspace/venv-train/lib/python3.13/site-packages/triton/profiler/scope.py__init__scope.__init__&s  cZ[R(dg[R"5Ul[R "URUR 5 UR(a?[5 [R"UR/[UR5Q76 ggr) r profiling_on libproton record_scoper enter_scoperrr add_metricsr rs r _enter_scopescope._enter_scope+si!! ((*dggtyy1 <<   ! !$'' S,DT\\,R S rc[R(a URcg[R"URUR 5 gr)r rrr exit_scoperr#s r _exit_scopescope._exit_scope4s.!!TWW_ TWWdii0rc&UR5 U$r)r$r#s r __enter__scope.__enter__9s  rc$UR5 gr)r()rexc_type exc_value tracebacks r__exit__scope.__exit__=s rc4^^[T5UU4Sj5nU$)Nc>TR5 T"U0UD6TR5 $!TR5 f=fr)r$r()argskwargsfuncrs rwrapperscope.__call__..wrapperBs:     #T,V,  "  "s+=r)rr7r8s`` r__call__scope.__call__@s t #  #r)rrrr)__name__ __module__ __qualname____firstlineno____doc__strrdictrrr$r(r+r1r:__static_attributes__rrrrsH,S8DcN+Ct T1  rrch^\rSrSrSrS S\S\\\\4SS4U4Sjjjr U4Sjr U4S jr S r U=r $) cpu_timed_scopeMz A scope that measures elapsed time (cpu_time). Args: name (str): The name of the scope. metrics (dict[str, float], optional): Additional metrics to add. Default is None. Nrrrcd>[TU]X5 SUlU(aSU;a [S5egg)Ncpu_timez'The metric name 'cpu_time' is reserved.)superr start_time ValueError)rrr __class__s rrcpu_timed_scope.__init__Vs4 ' zW,FG G-7rc>[R(dg[R"5Ul[ TU]5 gr)r rtimetime_nsrKrJr$)rrMs rr$cpu_timed_scope._enter_scope\s'!! ,,. rc>[R(dgURbE[R"5UR- n[ R "URSU05 [TU]%5 g)Nzcpu_time (ns)(exc)) r rrKrPrQrr"rrJr()rrIrMs rr(cpu_timed_scope._exit_scopebsR!!  ?? &||~7H  ! !$'',@(+K L r)rKr)r<r=r>r?r@rArrBrrr$r(rC __classcell__)rMs@rrFrFMsHHSH8DcN+CHtHH  rrF)rrrrc^[R(dg[R"5n[ [ S/5[ l[ R RX 45 [R"X 5 U(a*[5 [R"U/[U5Q76 U$)Nscopes) r rrr getattrthread_local_scopesrWappendr!r r"r rs rr!r!ksz       !B!()rgs 11@3oo'sHUOXc]BC;;|e<CG c xS#X'? 8TW=  RV Xc] xS#X7O [cdg[h r