sg1UddlZddlZddlmZddlmZddlmZmZddl m Z gdZ edgdZ Gd d Z Gd d ejZGd deZGddeZGddZiZeeefed<eaeed<d&dedeefdZdefdZdZd'defdZe ded&dZd(deded efd!Ze d"ed ededefd#Zd$e fd%Z!y))N) namedtuplewraps)DictOptional) deprecated) MetricsConfig MetricHandlerConsoleMetricHandlerNullMetricHandler MetricStream configure getStreamprofprofile put_metricpublish_metricget_elapsed_time_ms MetricDatar) timestamp group_namenamevaluec0eZdZdgZddeeeeffdZy)r paramsNc:||_|ji|_yyN)r)selfrs X/var/www/html/venv/lib/python3.12/site-packages/torch/distributed/elastic/metrics/api.py__init__zMetricsConfig.__init__(s ;; DK r)__name__ __module__ __qualname__ __slots__rrstrr r!rr r %s" IxS#X7r!r c6eZdZejdefdZy)r metric_datacyrr'rr)s remitzMetricHandler.emit/s r!N)r"r#r$abcabstractmethodrr,r'r!rr r .s!   r!r ceZdZdefdZy)r r)c td|jd|jd|jd|jy)N[z][z]: =)printrrrrr+s rr,zConsoleMetricHandler.emit5sD  %%&b)?)?(@KDTDTCUUVWbWhWhVi j r!Nr"r#r$rr,r'r!rr r 4s  r!r ceZdZdefdZy)r r)cyrr'r+s rr,zNullMetricHandler.emit<s r!Nr4r'r!rr r ;s  r!r c,eZdZdedefdZdedefdZy)r rhandlerc ||_||_yr)rr8)rrr8s rr zMetricStream.__init__As$ r! metric_name metric_valuec|jjttj|j||yr)r8r,rtimer)rr:r;s r add_valuezMetricStream.add_valueEs,  tyy{DOO[, O r!N)r"r#r$r&r r intr>r'r!rr r @s(3 S  r!r _metrics_map_default_metrics_handlerr8groupc ||ay|t|<yr)rAr@)r8rBs rrrPs }$+ % Ur!cJ|tvr t|}nt}t||Sr)r@rAr )rBr8s rrrZs& u%* w ''r!c|j}|jd}t|dk(r0|j}|r|jdddz|dzS|dS|S)N.r)r$splitlenr#)fnqualnamerImodules r_get_metric_namerNbs`H NN3 E 5zQ <<$R(3.q9 98Or!c&fd}|r||S|S)a @profile decorator publishes duration.ms, count, success, failure metrics for the function that it decorates. The metric name defaults to the qualified name (``class_name.def_name``) of the function. If the function does not belong to a class, it uses the leaf module name instead. Usage :: @metrics.prof def x(): pass @metrics.prof(group="agent") def y(): pass c2tfd}|S)Nc$t} tj}|i|}t|dd t|dt ||S#t$rt|ddwxYw#t|dt wxYwNz.successrGz.failurez .duration.ms)rNr=r Exceptionr)argskwargskeystartresultfrBs rwrapperz#prof..wrap..wrappers"1%C T D+F+cU(+Q6 cU,/1DU1KUSM  cU(+Q6 cU,/1DU1KUSs,AA11A44Br)rYrZrBs` rwrapzprof..wraps q   r!r')rKrBr[s ` rrros(" Bx r!zDeprecated, use `@prof` instead)categorycfd}|S)z @profile decorator adds latency and success/failure metrics to any given function. Usage :: @metrics.profile("my_metric_group") def some_function(): c2tfd}|S)Nc ^ tj}|i|}tjdd tjdt ||S#t$rtjddwxYw#tjdt wxYwrR)r=rr"rSr)rTrU start_timerXfuncrBs rrZz&profile..wrap..wrappers !YY[ t.v.ux&@!D }}o\2' 3 M ux&@!D }}o\2' 3s6A%BB%B,r)rarZrBs` rr[zprofile..wraps t    r!r')rBr[s` rrrs( Kr!r:r; metric_groupc:t|j||y)z Publish a metric data point. Usage :: put_metric("metric_name", 1) put_metric("metric_name", 1, "metric_group_name") Nrr>)r:r;rbs rrrsl%%k<@r!zMDeprecated, use `put_metric(metric_group)(metric_name, metric_value)` insteadc>t|}|j||yrrd)rbr:r; metric_streams rrrs l+MK6r!start_time_in_secondscLtj}t||z dzS)zrts@ "!(   &R S  CGG  =     *, d3 %&+*;*=-=&}&Xc]&(S( ((V - F G F AC As A# A S 7737c7 7 :u:r!