U ʹj} @sPUddlZddlZddlZddlmZddlmZddlmZm Z ddddd d d d d ddddg Z edddddgZ GdddZ Gdddej ZGdddeZGdddeZGdd d ZiZeaeed<d+ee eddd Zeddd Zd d!Zd,edd#d Zd-d$d Zd.eeed%d&dZeeed'd(dZed)d*dZdS)/N) namedtuplewraps)DictOptional MetricsConfig MetricHandlerConsoleMetricHandlerNullMetricHandler MetricStream configure getStreamprofprofile put_metricpublish_metricget_elapsed_time_ms MetricData timestamp group_namenamevaluec@s.eZdZdgZdeeeefdddZdS)rparamsNrcCs||_|jdkri|_dSNr)selfrrd/var/www/html/bwave/python/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)rr r!abcabstractmethodrr'rrrrr sc@seZdZedddZdS)r r$c Cs,td|jd|jd|jd|jdS)N[z][z]: =)printrrrrr&rrrr''s"zConsoleMetricHandler.emitNrr r!rr'rrrrr &sc@seZdZedddZdS)r r$cCsdSrrr&rrrr'.szNullMetricHandler.emitNr-rrrrr -sc@s,eZdZeedddZeedddZdS)r rhandlercCs||_||_dSrr.)rrr/rrrr3szMetricStream.__init__) metric_name metric_valuecCs |jtt|j||dSr)r/r'rtimer)rr0r1rrr add_value7szMetricStream.add_valueN)rr r!r#rrintr3rrrrr 2s_default_metrics_handlerr/groupcCs|dkr|an|t|<dSr)r5 _metrics_mapr6rrrr Bsr7cCs |tkrt|}nt}t||Sr)r8r5r )r7r/rrrr Ls cCsR|j}|d}t|dkrJ|j}|r@|ddd|dS|dSn|SdS)N.r)r!splitlenr )fnqualnamer=modulerrr_get_metric_nameTs   rB 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)rBrrr2 Exception)argskwargskeystartresult)fr7rrwrapperws  z#prof..wrap..wrapperr)rKrLr9)rKrwrapvs zprof..wrapNr)r?r7rMrr9rras cstdtfdd}|S)z @profile decorator adds latency and success/failure metrics to any given function. Usage :: @metrics.profile("my_metric_group") def some_function(): zDeprecated, use @prof insteadcstfdd}|S)Nc szzZz*t}||}tjddWn*tk rVtjddYnXW5tjdt|X|SrD)rrrr2rE)rFrG start_timerJ)funcr7rrrLs   z&profile..wrap..wrapperr)rOrLr9)rOrrMszprofile..wrap)warningswarnDeprecationWarning)r7rMrr9rrs  r0r1 metric_groupcCst|||dS)z Publishes a metric data point. Usage :: put_metric("metric_name", 1) put_metric("metric_name", 1, "metric_group_name") N)r r3rSrrrrs )rTr0r1cCs"tdt|}|||dS)NzKDeprecated, use put_metric(metric_group)(metric_name, metric_value) instead)rPrQr r3)rTr0r1Z metric_streamrrrrs )start_time_in_secondscCst}t||dS)zG Returns the elapsed time in millis from the given start time. i)r2r4)rUZend_timerrrrs)N)NrC)N)rC) r(r2rP collectionsr functoolsrtypingrr__all__rrABCrr r r r8r5__annotations__r#r r rBrrr4rrfloatrrrrr s@       , $