a h,@s^UddlZddlZddlmZddlmZddlmZddlm Z gdZ edgdZ Gd d d Z Gd d d ej ZGd ddeZGdddeZGdddZiZeeefed<eaeed<d.eeedddZedddZddZd/edddZe d ed!d0d"d#Zd1eeed$d%d&Ze d'ed!eeed(d)d*Ze d+d,d-Z!dS)2N) namedtuplewraps)Optional) deprecated) MetricsConfig MetricHandlerConsoleMetricHandlerNullMetricHandler MetricStream configure getStreamprofprofile put_metricpublish_metricget_elapsed_time_ms MetricDatar) timestamp group_namenamevaluec@s.eZdZdgZdeeeefdddZdS)rparamsNrcCs||_|jduri|_dSNr)selfrrS/var/www/auris/lib/python3.9/site-packages/torch/distributed/elastic/metrics/api.py__init__(s zMetricsConfig.__init__)N)__name__ __module__ __qualname__ __slots__rdictstrrrrrrr%src@s eZdZejedddZdS)r metric_datacCsdSrrrr&rrremit/szMetricHandler.emitN)rr r!abcabstractmethodrr(rrrrr.src@seZdZedddZdS)r r%c Cs,td|jd|jd|jd|jdS)N[z][z]: =)printrrrrr'rrrr(5s"zConsoleMetricHandler.emitNrr r!rr(rrrrr 4sr c@seZdZedddZdS)r r%cCsdSrrr'rrrr(<szNullMetricHandler.emitNr.rrrrr ;sr c@s,eZdZeedddZeedddZdS)r rhandlercCs||_||_dSrr/)rrr0rrrrAszMetricStream.__init__) metric_name metric_valuecCs |jtt|j||dSr)r0r(rtimer)rr1r2rrr add_valueEszMetricStream.add_valueN)rr r!r$rrintr4rrrrr @sr _metrics_map_default_metrics_handlerr0groupcCs|dur|an|t|<dSr)r7r6r8rrrr Psr r9cCs |tvrt|}nt}t||Sr)r6r7r )r9r0rrrr Zs r cCsR|j}|d}t|dkrJ|j}|r@|ddd|dS|dSn|SdS)N.r)r!splitlenr )fnqualnamer>modulerrr_get_metric_namebs   rC torchelasticcs fdd}|r||S|SdS)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 cstfdd}|S)Nc st}znz,t}|i|}t|ddWn&ty\t|ddYn0Wt|dt|nt|dt|0|SNz.successr<z.failurez .duration.ms)rCr3r Exceptionr)argskwargskeystartresult)fr9rrwrappers  0z#prof..wrap..wrapperr)rLrMr:)rLrwraps zprof..wrapNr)r@r9rNrr:rros rzDeprecated, use `@prof` instead)categorycsfdd}|S)z @profile decorator adds latency and success/failure metrics to any given function. Usage :: @metrics.profile("my_metric_group") def some_function(): cstfdd}|S)Nc sztz.t}|i|}tjddWn(tyXtjddYn0Wtjdt|ntjdt|0|SrE)r3rrrFr)rGrH start_timerK)funcr9rrrMs"    z&profile..wrap..wrapperr)rQrMr:)rQrrNszprofile..wrapr)r9rNrr:rrs rr1r2 metric_groupcCst|||dS)z Publish a metric data point. Usage :: put_metric("metric_name", 1) put_metric("metric_name", 1, "metric_group_name") Nr r4rRrrrrs rzMDeprecated, use `put_metric(metric_group)(metric_name, metric_value)` instead)rSr1r2cCst|}|||dSrrT)rSr1r2Z metric_streamrrrrsr)start_time_in_secondscCst}t||dS)z s6         ,  #