IЦi%SSKrSSKJrJrJrJrJr SSKJr \\ \ \\ \4\\\ \\ /S4r \\ S'"SS5rg)N)AnyCallableDictOptionalType) TypeAlias OnExitTypec\rSrSrS\4SjrSSjrS\\\ S\\ S\ SS 4S jr S\ 4S jr S \S \SS 4SjrS \S \ SS 4SjrS \S\S \ SS 4SjrS\\\ 4SS 4SjrS\\\ 4SS 4SjrS \S \ SS 4SjrSrg )MetricsContext on_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_level)selfr s \/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/torch/_dynamo/metrics_context.py__init__MetricsContext.__init__ s (* #$ returncURS:Xa!0Ul[R"5UlU=RS- slU$)z Initialize metrics recording. r)rrtimetime_nsrrs 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. rrN)rrrrrr)rr r!r" end_time_nss r__exit__MetricsContext.__exit__%s] q {{a ;;! ,,.K MM##[--  rc URS:$)z$ True if we've entered the context. r)rrs r in_progressMetricsContext.in_progress6s{{QrmetricvaluecURS:Xa[SUS35eXR;aSURU'URU==U- ss'g)z' Increment a metric by a given amount. rzCannot increment  outside of a MetricsContextNr RuntimeErrorrrr*r+s r incrementMetricsContext.increment<sR ;;! !26(:VWX X  &$%DMM& ! f&rcURS:Xa[SUS35eXR;a[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 r-zMetric 'z-' has already been set in the current contextNr.r0s rsetMetricsContext.setFsW ;;! VH4PQR R ]] "6("OP !& frkeycURS: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). rr4r-Nr.)rr*r7r+s r set_key_valueMetricsContext.set_key_valueSsN ;;! VH4PQR R  &$&DMM& !%* fc"rvaluescURS:Xa [S5eURR5UR5-nU(a[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. r1Cannot update metrics outside of a MetricsContextz Metric(s) z- have already been set in the current contextN)rr/rkeysupdate)rr;existings rr?MetricsContext.updateasj ;;! RS S==%%'&++-7 XJ&ST  V$rc~URS:Xa [S5eURS:XaURU5 gg)z1 Update, but only when at the outermost context. rr=rN)rr/r?)rr;s r update_outerMetricsContext.update_outeros9 ;;! RS S ;;!  KK  rcURS:Xa[SUS35eXR;a[5URU'URUR U5 g)z( Records a metric as a set() of values. rz Cannot add r-N)rr/rr5addr0s r add_to_setMetricsContext.add_to_setxsV ;;! VH4PQR R  &$'EDMM& ! f!!%(r)rrrr)rr )__name__ __module__ __qualname____firstlineno__r rrrr BaseExceptionrr%boolr(strintr1r5r9rr?rCrG__static_attributes__rrr r s   4 ./M*   "T ''C'D' &# &c &d & +C +c +# +$ + %T#s(^ % % 4S> d ))S)T)rr )rtypingrrrrrtyping_extensionsrrPrOrMr __annotations__r rRrrrVs] 66'!#tCH~x](;UV  I t)t)r