U yh @s|UddlZddlZddlmZddlmZddlmZmZddl m Z dddd d d d d dddddg Z edddddgZ GdddZ GdddejZGdddeZGdd d eZGdd d ZiZeeefed<eaeed<d0eeeddd Zed d!d Zd"d#Zd1ed d%d Ze d&ed'd2d(dZd3eeed)d*dZe d+ed'eeed,d-dZe d.d/dZ!dS)4N) namedtuplewraps)DictOptional) deprecated MetricsConfig MetricHandlerConsoleMetricHandlerNullMetricHandler MetricStream configure getStreamprofprofile put_metricpublish_metricget_elapsed_time_ms MetricData timestamp group_namenamevaluec@s.eZdZdgZdeeeefdddZdS)rparamsNrcCs||_|jdkri|_dSNr)selfrrW/var/www/html/venv/lib/python3.8/site-packages/torch/distributed/elastic/metrics/api.py__init__s zMetricsConfig.__init__)N)__name__ __module__ __qualname__ __slots__rrstrrrrrrrsc@s eZdZejedddZdS)r  metric_datacCsdSrrrr&rrremit"szMetricHandler.emitN)r r!r"abcabstractmethodrr(rrrrr !sc@seZdZedddZdS)r r%c Cs,td|jd|jd|jd|jdS)N[z][z]: =)printrrrrr'rrrr((s"zConsoleMetricHandler.emitNr r!r"rr(rrrrr 'sc@seZdZedddZdS)r r%cCsdSrrr'rrrr(/szNullMetricHandler.emitNr.rrrrr .sc@s,eZdZeedddZeedddZdS)r rhandlercCs||_||_dSrr/)rrr0rrrr4szMetricStream.__init__) metric_name metric_valuecCs |jtt|j||dSr)r0r(rtimer)rr1r2rrr add_value8szMetricStream.add_valueN)r r!r"r$r rintr4rrrrr 3s _metrics_map_default_metrics_handlerr0groupcCs|dkr|an|t|<dSr)r7r6r8rrrr Csr9cCs |tkrt|}nt}t||Sr)r6r7r )r9r0rrrrMs cCsR|j}|d}t|dkrJ|j}|r@|ddd|dS|dSn|SdS)N.r)r"splitlenr!)fnqualnamer>modulerrr_get_metric_nameUs   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 s|t}zVz(t}||}t|ddWn(tk rZt|ddYnXW5t|dt|X|SNz .duration.msz.successr<z.failure)rCrrr3 Exception)argskwargskeystartresult)fr9rrwrapperws  z#prof..wrap..wrapperr)rLrMr:)rLrwrapvs zprof..wrapNr)r@r9rNrr:rrbs zDeprecated, 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 szzZz*t}||}tjddWn*tk rVtjddYnXW5tjdt|X|SrE)rr rr3rF)rGrH start_timerK)funcr9rrrMs   z&profile..wrap..wrapperr)rQrMr:)rQrrNszprofile..wrapr)r9rNrr:rrs r1r2 metric_groupcCst|||dS)z Publish a metric data point. Usage :: put_metric("metric_name", 1) put_metric("metric_name", 1, "metric_group_name") Nrr4rRrrrrs zMDeprecated, use `put_metric(metric_group)(metric_name, metric_value)` instead)rSr1r2cCst|}|||dSrrT)rSr1r2Z metric_streamrrrrs)start_time_in_secondscCst}t||dS)z sJ        +  "