`i,UddlZddlZddlmZddlmZddlmZddlm Z gdZ edgdZ Gd d Z Gd d ej ZGd deZGddeZGddZiZeeefed<eaeed<d'dedeefdZdefdZdZd(defdZe ded'dZd)ded ed!efd"Ze d#ed!eded efd$Zd%e fd&Z!dS)*N) namedtuplewraps)Optional) deprecated) MetricsConfig MetricHandlerConsoleMetricHandlerNullMetricHandler MetricStream configure getStreamprofprofile put_metricpublish_metricget_elapsed_time_ms MetricDatar) timestamp group_namenamevaluec>eZdZdgZddeeeeffdZdS)rparamsNc4||_|j i|_dSdSN)r)selfrs y/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torch/distributed/elastic/metrics/api.py__init__zMetricsConfig.__init__(s$ ; DKKK  r)__name__ __module__ __qualname__ __slots__rdictstrrr rrr%sH IxS#X7r rc4eZdZejdefdZdS)r metric_datacdSrr'rr)s remitzMetricHandler.emit/s r N)r!r"r#abcabstractmethodrr,r'r rr r .s=       r r ceZdZdefdZdS)r r)c dtd|jd|jd|jd|jdS)N[z][z]: =)printrrrrr+s rr,zConsoleMetricHandler.emit5sN j % j j)? j jKDT j jWbWh j j     r Nr!r"r#rr,r'r rr r 4s/       r r ceZdZdefdZdS)r r)cdSrr'r+s rr,zNullMetricHandler.emit<s r Nr4r'r rr r ;s/       r r c.eZdZdedefdZdedefdZdS)r rhandlerc"||_||_dSr)rr8)rrr8s rrzMetricStream.__init__As$ r metric_name metric_valuec|jttj|j||dSr)r8r,rtimer)rr:r;s r add_valuezMetricStream.add_valueEs?  ty{{DO[, O O     r N)r!r"r#r&r rintr>r'r rr r @sX3 S       r r _metrics_map_default_metrics_handlerr8groupc&||adS|t|<dSr)rAr@)r8rBs rr r Ps% }$+   % Ur c^|tvrt|}nt}t||Sr)r@rAr )rBr8s rrrZs. u%* w ' ''r c|j}|d}t|dkr8|j}|r'|dddz|dzS|dS|S)N.r)r#splitlenr")fnqualnamerImodules r_get_metric_namerNbskH NN3  E 5zzQ  <<$$R(3.q9 98Or torchelasticc,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 c@tfd}|S)Nc`t} tj}|i|}t|ddn##t$rt|ddwxYw t|dt |n&#t|dt |wxYw|SNz.successrGz.failurez .duration.ms)rNr=r Exceptionr)argskwargskeystartresultfrBs rwrapperz#prof..wrap..wrappers"1%%C T D+F++c+++Q6666   c+++Q666 7 c///1DU1K1KUSSSS c///1DU1K1KUSSSSMs/AB A""B#B+r)rZr[rBs` rwrapzprof..wraps9 q       r r')rKrBr\s ` rrros8*" tBxx 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(): c@tfd}|S)Nc j tj}|i|}tjddn(#t$rtjddwxYw tjdt |n+#tjdt |wxYw|SrS)r=rr!rTr)rUrV start_timerYfuncrBs rr[z&profile..wrap..wrappers !Y[[ t.v..u&@&@&@!DDDD   u&@&@&@!DDD E }222' 33}222' 33 Ms48B%AB(B0r)rbr[rBs` rr\zprofile..wraps9 t        r r')rBr\s` rrrs#( Kr r:r; metric_groupcLt|||dS)z Publish a metric data point. Usage :: put_metric("metric_name", 1) put_metric("metric_name", 1, "metric_group_name") Nrr>)r:r;rcs rrrs(l%%k<@@@@@r zMDeprecated, use `put_metric(metric_group)(metric_name, metric_value)` insteadcPt|}|||dSrre)rcr:r; metric_streams rrrs- l++MK66666r start_time_in_secondscRtj}t||z dzS)zrts """"""((((((    Z &R&R&R S S      CG        =                   *, d3 %&+++*;*;*=*=-===&&}&Xc]&&&&(S((((   ))))))X  - FFF   GF F A AC As A# A A A A S 7737c777 7 :u::::::r