+ j#Pa0t$Rt^RIHt^RIt^RIt^RIt^RIHt^RI H t H t H t ^RI Ht] 'd^RIHt^RIHt]P$!]4t!RR 4t]]]]]] 3,]],R,]R,.R3,tR ]R &!R R 4t!RR4tR#)a[Metrics collection and management system for Dynamo. This module provides context managers for gathering and reporting metrics during compilation and runtime. It includes two main components: - MetricsContext: A context manager for collecting metrics during compilation, supporting nested contexts and various metric types (counters, sets, key-value pairs) - RuntimeMetricsContext: A specialized context for runtime metrics collection that doesn't require explicit context management The metrics system enables comprehensive monitoring and analysis of both compilation and execution performance. ) annotationsN)Callable)Any TYPE_CHECKING TypeAlias)Self)Iterator)CapturedTracebackcN]tRt^#tRtR RRlltRRltRRltRR ltR t R #) TopNzW Helper to record a list of metrics, keeping only the top N "most expensive" elements. c V^8dQhRRRR/#)at_mostintreturnNone)formats"^/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/_dynamo/metrics_context.py __annotate__TopN.__annotate__(s...T.c Wn.VnR#Nrheap)selfrs&&r__init__ TopN.__init__(s +- rc$V^8dQhRRRRRR/#)r keyrvalrrrr)rs"rrr,s!""s"""rc \VP4VP8d\PM\P pV!VPW!34R#r)lenrrheapqheappush heappushpop)rr r!fns&&& raddTopN.add,s6"499~ R\VPRR74#)c3,"TF wrW!3xK R#5irr).0r!r s& r TopN.__iter__..5sK+Jxs +JsT)reverse)sortedrr,s&r__iter__ TopN.__iter__4sK6$))T+JKKrrN)) __name__ __module__ __qualname____firstlineno____doc__rr(r-r7__static_attributes__rrrr r #s"." LLrr r OnExitTypec]tRt^>tRRltRRltRRltRRltR R ltR R lt RR Rllt RRlt RRRllt RRlt RRltRRltRtR#)MetricsContextc V^8dQhRRRR/#r on_exitr@rrr)rs"rrMetricsContext.__annotate__?s C C Ct Crc JWn/Vn^Vn^Vn.VnR#)a) Use this class as a contextmanager to create a context under which to accumulate a set of metrics, e.g., metrics gathered during a compilation. On exit of the contextmanager, call the provided 'on_exit' function and pass a dictionary of all metrics set during the lifetime of the contextmanager. N)_on_exit_metrics_start_time_ns_level_editsrrEs&&rrMetricsContext.__init__?s& (* #$ @B rcV^8dQhRR/#)r rrr)rs"rrrFLs  4 rc VP^8Xd"/Vn\P!4VnV;P^, unV#)z Initialize metrics recording. )rKrItimetime_nsrJr,s&r __enter__MetricsContext.__enter__Ls8 ;;! DM"&,,.D  q  rc(V^8dQhRRRRRRRR/#) r exc_typeztype[BaseException] | None exc_valuezBaseException | None _tracebackrrrr)rs"rrrFXs8RR,R(R R  Rrc ^V;P^,unVP^8d \R4hVP^8Xd@\P!4pVP VP W@P W4R#R# \d\PR4R#i;i)z. At exit, call the provided on_exit function. z+MetricsContext level cannot become negativez0Unexpected exception logging compilation metricsN) rKAssertionErrorrQrRrHrJrI Exceptionlog exception)rrVrWrX end_time_nss&&&& r__exit__MetricsContext.__exit__Xs q ;;? !NO O ;;!  R"lln  ''mmX   R PQ RswrVS,'gKRRPVP44,xK@ R#5i)zPrevious Traceback: N)joinr)r2ekrss& rr3/MetricsContext._render_edits..~s=$ #4xx : #bggahhj&9 9 9#s A .A )rwrL)rrss&fr _render_editsMetricsContext._render_edits}s, $  $    rc(V^8dQhRRRRRRRR/#) r rgrhrir overwriterbrrr)rs"rrrFs(&&#&c&d&t&rc DVP^8Xd\RV R24hWP9d/V'g'\VPV04RV R2,4hVPP \ P!^R7V034W PV&R#)zj Set a metric to a given value. Raises if the metric has been assigned previously in the current context. Cannot set rlz RuntimeError: Metric 'z]' has already been set in the current context (see above for current and previous traceback).skipN)rKrnrIr{rLappendr extract)rrgrir~s&&&&rsetMetricsContext.sets ;;! VH4PQR R ]] "9""F8,.vh7BBB  -551=xHI % frc(V^8dQhRRRRRRRR/#)r rgrhr rirrrr)rs"rrrFs( + +C +c +# +$ +rc VP^8Xd\RV R24hWP9d/VPV&W0PV,V&R#)a Treats a given metric as a dictionary and set the k and value within it. Note that the metric must be a dictionary or not present. We allow this to be called multiple times (i.e. for features, it's not uncommon for them to be used multiple times within a single compilation). rrlNrm)rrgr ris&&&&r set_key_valueMetricsContext.set_key_valuesN ;;! VH4PQR R  &$&DMM& !%* fc"rc$V^8dQhRRRRRR/#)r valuesdict[str, Any]r~rbrrr)rs"rrrFs!%%^%%%rc VP^8Xd \R4hVPP4VP4,pV'dEV'g=\VP \ VP444RV R2,4hVP P\P!^R7\ VP4434VPPV4R#)z Set multiple metrics directly. This method does NOT increment. Raises if any metric has been assigned previously in the current context and overwrite is not set to True. 1Cannot update metrics outside of a MetricsContextz RuntimeError: Metric(s) z_ have already been set in the current context. (see above for current and previous traceback).rN) rKrnrIkeysr{rrLrr rupdate)rrr~existings&&& rrMetricsContext.updates ;;! RS S==%%'&++-7 I""3v{{}#560 ;BBB  -551=s6;;=?QRS V$rc V^8dQhRRRR/#)r rrrrr)rs"rrrFs  > d rc VP^8Xd \R4hVP^8XdVPV4R#R#)z1 Update, but only when at the outermost context. rN)rKrnr)rrs&&r update_outerMetricsContext.update_outers9 ;;! RS S ;;!  KK  rc$V^8dQhRRRRRR/#)r rgrhrirrrr)rs"rrrFs!)))S)T)rc VP^8Xd\RV R24hWP9d\4VPV&VPV,P V4R#)z( Records a metric as a set() of values. z Cannot add rlN)rKrnrIrr(ros&&&r add_to_setMetricsContext.add_to_setsV ;;! VH4PQR R  &$'EDMM& ! f!!%(rc(V^8dQhRRRRRRRR/#) r rgrhr rr!rrrr)rs"rrrFs(,,,#,C,D,rc VP^8XdR#WP9d\4VPV&VPV,PW#4R#)z+ Records a metric as a TopN set of values. N)rKrIr r()rrgr r!s&&&&r add_top_nMetricsContext.add_top_nsD ;;!    &$(FDMM& ! f!!#+r)rLrKrIrHrJN)F)r:r;r<r=rrSr_rdrpr{rrrrrrr?rrrrBrB>sE C R* ' & +%$ ),,rrBc>]tRt^tRRltR RRlltRRltRtR#) RuntimeMetricsContextc V^8dQhRRRR/#rDr)rs"rr"RuntimeMetricsContext.__annotate__s%% %t%rc .Wn/Vn^VnR#)z Similar to MetricsContext, but used to gather the runtime metrics that are decoupled from compilation, where there's not a natural place to insert a context manager. N)rHrIrJrMs&&rrRuntimeMetricsContext.__init__s  (* #$rNc(V^8dQhRRRRRRRR/#) r rgrhrirextrazdict[str, Any] | Nonerrr)rs"rrrs,)))"%).C) )rc lVP'g\P!4VnWP9d^VPV&VPV;;,V, uu&V'dBVP 4F+wrEW@P9gKVfKWPPV&K- R#R#)rkN)rIrQrRrJitems)rrgrirryvs&&&& rrpRuntimeMetricsContext.increments{ }}}"&,,.D   &$%DMM& ! f&  MM)am'(MM!$& rcV^8dQhRR/#)r rrr)rs"rrrs # # #rc VP'dH\P!4pVPVPWPRR4/VnR#R# \ d\ PR4L-i;i /Tni;i)zG Call the on_exit function with the metrics gathered so far and reset. Nz,Unexpected exception logging runtime metrics)rIrQrRrHrJr[r\r])rr^s& rfinishRuntimeMetricsContext.finishst === #"lln  ''mmT4 !#    N LM N!# s#=A A?<B>A??B B )rIrHrJr)r:r;r<r=rrprr?rrrrrs%)$ # #rr)__conditional_annotations__r> __future__rr$loggingrQcollections.abcrtypingrrrtyping_extensionsrrtorch.utils._tracebackr getLoggerr:r\r rdictrhtype BaseExceptionr@__annotations__rBr)rs@rrs #  $00"(4!LL*!#tCH~tM2T9=4;OP  I Q,Q,h*#*#r