g^(h,UddlZddlZddlmZddlmZddlmZddlm 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)Optional) deprecated) MetricsConfig MetricHandlerConsoleMetricHandlerNullMetricHandler MetricStream configure getStreamprofprofile put_metricpublish_metricget_elapsed_time_ms MetricDatar) timestamp group_namenamevaluec0eZdZdgZddeeeeffdZy)rparamsNc:||_|ji|_yyN)r)selfrs c/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/torch/distributed/elastic/metrics/api.py__init__zMetricsConfig.__init__(s ;; DK r)__name__ __module__ __qualname__ __slots__rdictstrrr rrr%s" IxS#X7r rc6eZdZejdefdZy)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 rrzMetricStream.__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 rintr>r'r rr r @s(3 S  r r _metrics_map_default_metrics_handlerr8groupc ||ay|t|<yr)rAr@)r8rBs rr r Ps }$+ % 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( ))X - F G F AC As A# A S 7737c7 7 :u:r