[Th%SrSSKrSSKrSSKJr SSKJrJrJr SSK J r "SS5r \\ \ \ \\4\\\\\/S4r\ \S'"S S 5r"S S 5rg) 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. N)Iterator)AnyCallableOptional) TypeAliascl\rSrSrSrS S\4SjjrS\S\SS4S jrS\4S jr S\ \ \\44S jr S r g)TopNzW Helper to record a list of metrics, keeping only the top N "most expensive" elements. at_mostcXl/UlgNr heap)selfr s U/var/www/auris/envauris/lib/python3.13/site-packages/torch/_dynamo/metrics_context.py__init__ TopN.__init__s +- keyvalreturnNc[UR5UR:a[RO[R nU"URX!45 gr )lenrr heapqheappush heappushpop)rrrfns raddTopN.add s6"499~  TopN.__iter__..)sK+Jxs +JsT)reverse)sortedrr!s r__iter__ TopN.__iter__(sK6$))T+JKKrr))__name__ __module__ __qualname____firstlineno____doc__intrrrr"rtupler,__static_attributes__r&rrr r sT.."s""" L(5c?3Lrr OnExitTypec $\rSrSrS\4SjrSSjrS\\\ S\\ S\ SS 4S jr S\ 4S jr S \S \SS 4SjrSS \S \ S\ SS 4SjjrS \S\S \ SS 4SjrSS\\\ 4S\ SS 4SjjrS\\\ 4SS 4SjrS \S \ SS 4SjrS \S\ S\SS 4SjrSrg )MetricsContext2on_exitc:Xl0UlSUlSUlg)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. rN)_on_exit_metrics_start_time_ns_levelrr;s rrMetricsContext.__init__3s (* #$ rrcURS:Xa!0Ul[R"5UlU=RS- slU$)z Initialize metrics recording. r)r@r>timetime_nsr?r!s r __enter__MetricsContext.__enter__?s8 ;;! DM"&,,.D  q  rexc_type exc_value _tracebackNcU=RS-slURS:deURS:Xa=[R"5nURURX@R X5 gg)z. At exit, call the provided on_exit function. rDrN)r@rErFr=r?r>)rrIrJrK end_time_nss r__exit__MetricsContext.__exit__Ks] q {{a ;;! ,,.K MM##[--  rc URS:$)z$ True if we've entered the context. r)r@r!s r in_progressMetricsContext.in_progress\s{{QrmetricvaluecURS:Xa[SUS35eXR;aSURU'URU==U- ss'g)' Increment a metric by a given amount. rzCannot increment  outside of a MetricsContextNr@ RuntimeErrorr>rrSrTs r incrementMetricsContext.incrementbsR ;;! !26(:VWX X  &$%DMM& ! f&r overwritecURS:Xa[SUS35eXR;aU(d[SUS35eX RU'g)zj Set a metric to a given value. Raises if the metric has been assigned previously in the current context. r Cannot set rWzMetric 'z-' has already been set in the current contextNrX)rrSrTr]s rsetMetricsContext.setlsY ;;! VH4PQR R ]] "96("OP !& frrcURS:Xa[SUS35eXR;a0URU'X0RUU'g)a Treats a give 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). rr_rWNrX)rrSrrTs r set_key_valueMetricsContext.set_key_valueysN ;;! VH4PQR R  &$&DMM& !%* fc"rvaluescURS:Xa [S5eURR5UR5-nU(aU(d[SUS35eURR U5 g)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. r1Cannot update metrics outside of a MetricsContextz Metric(s) z- have already been set in the current contextN)r@rYr>keysupdate)rrer]existings rriMetricsContext.updatesl ;;! RS S==%%'&++-7 IXJ&ST  V$rc~URS:Xa [S5eURS:XaURU5 gg)z1 Update, but only when at the outermost context. rrgrDN)r@rYri)rres r update_outerMetricsContext.update_outers9 ;;! RS S ;;!  KK  rcURS:Xa[SUS35eXR;a[5URU'URUR U5 g)z( Records a metric as a set() of values. rz Cannot add rWN)r@rYr>r`rrZs r add_to_setMetricsContext.add_to_setsV ;;! VH4PQR R  &$'EDMM& ! f!!%(rrcURS:XagXR;a[5URU'URURX#5 g)z+ Records a metric as a TopN set of values. rN)r@r>r r)rrSrrs r add_top_nMetricsContext.add_top_nsD ;;!    &$(FDMM& ! f!!#+r)r@r>r=r?)rr9)F)r/r0r1r2r7rrGrtype BaseExceptionrrNboolrQstrr4r[r`rcdictrirmrprsr6r&rrr9r92s-   4 ./M*   "T ''C'D' &# &c &d &t & +C +c +# +$ + %T#s(^ % % % 4S> d ))S)T),,#,C,D,rr9c V\rSrSrS\4SjrS\S\S\\ \\ 4SS4S jr S S jr S r g) RuntimeMetricsContextr;c,Xl0UlSUlg)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. rN)r=r>r?rAs rrRuntimeMetricsContext.__init__s  (* #$rrSrTextrarNcNUR(d[R"5UlXR;aSURU'URU==U- ss'U(a>UR 5H)upEX@R;dMUcMXPRU'M+ gg)rVrN)r>rErFr?items)rrSrTrkvs rr[RuntimeMetricsContext.incrementsy }}"&,,.D   &$%DMM& ! f&  MM)am'(MM!$& rcUR(aE[R"5nURURXRSS5 0Ulgg)zG Call the on_exit function with the metrics gathered so far and reset. N)r>rErFr=r?)rrMs rfinishRuntimeMetricsContext.finishs? ==,,.K MM$--{MM4QU VDM r)r>r=r?)rN)r/r0r1r2r7rrxr4rryrr[rr6r&rrr{r{sE% %))"%).6tCH~.F) )$rr{)r3rrEcollections.abcrtypingrrrtyping_extensionsrr r4ryrxrurvr7__annotations__r9r{r&rrrs  $**'LL*!#tCH~x](;UV  I ,,D$$r