U ‰dõã@s.UddlZddlZddlZddlmZddlmZddlmZm Z eddddd gƒZ Gd d „d ƒZ Gd d „d ej ƒZ Gdd„de ƒZGdd„de ƒZGdd„dƒZiZeƒae ed<d+e edœdd„Zedœdd„Zdd„Zd,edœdd„Zd-d d!„Zd.eeed"œd#d$„Zeeed%œd&d'„Zed(œd)d*„ZdS)/éN)Ú namedtuple©Úwraps)ÚDictÚOptionalÚ MetricDataÚ timestampÚ group_nameÚnameÚvaluec@s.eZdZdgZdeeeefdœdd„ZdS)Ú MetricsConfigÚparamsN©r cCs||_|jdkri|_dS©Nr)Úselfr ©rúI/tmp/pip-unpacked-wheel-ua33x9lu/torch/distributed/elastic/metrics/api.pyÚ__init__s zMetricsConfig.__init__)N)Ú__name__Ú __module__Ú __qualname__Ú __slots__rrÚstrrrrrrr sr c@s eZdZejedœdd„ƒZdS)Ú MetricHandler©Ú metric_datacCsdSrr©rrrrrÚemitszMetricHandler.emitN)rrrÚabcÚabstractmethodrrrrrrrsrc@seZdZedœdd„ZdS)ÚConsoleMetricHandlerrcCs td |j|j|j|j¡ƒdS)Nz[{}][{}]: {}={})ÚprintÚformatrr r r rrrrr#süÿzConsoleMetricHandler.emitN©rrrrrrrrrr "sr c@seZdZedœdd„ZdS)ÚNullMetricHandlerrcCsdSrrrrrrr/szNullMetricHandler.emitNr#rrrrr$.sr$c@s,eZdZeedœdd„Zeedœdd„ZdS)Ú MetricStream©r ÚhandlercCs||_||_dSrr&)rr r'rrrr4szMetricStream.__init__)Ú metric_nameÚ metric_valuecCs |j tt ¡|j||ƒ¡dSr)r'rrÚtimer )rr(r)rrrÚ add_value8sÿzMetricStream.add_valueN)rrrrrrÚintr+rrrrr%3sr%Ú_default_metrics_handler©r'ÚgroupcCs|dkr|an|t|<dSr)r-Ú _metrics_mapr.rrrÚ configureCsr1©r/cCs |tkrt|}nt}t||ƒSr)r0r-r%)r/r'rrrÚ getStreamMs r3cCsR|j}| d¡}t|ƒdkrJ|j}|r@| d¡dd|dS|dSn|SdS)NÚ.ééÿÿÿÿr)rÚsplitÚlenr)ÚfnÚqualnamer7ÚmodulerrrÚ_get_metric_nameUs   r<Ú 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 cstˆƒ‡‡fdd„ƒ}|S)Nc s|tˆƒ}zVz(t ¡}ˆ||Ž}t|›ddˆƒWn(tk rZt|›ddˆƒ‚YnXW5t|›dt|ƒˆƒX|S)Nz .duration.msz.successr5z.failure)r<Ú put_metricÚget_elapsed_time_msr*Ú Exception)ÚargsÚkwargsÚkeyÚstartÚresult)Úfr/rrÚwrapperxs  z#prof..wrap..wrapperr)rFrGr2)rFrÚwrapws zprof..wrapNr)r9r/rHrr2rÚprofbs rIcst dt¡‡fdd„}|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 insteadcstˆƒ‡‡fdd„ƒ}|S)Nc szzZz*t ¡}ˆ||Ž}tˆd ˆj¡dƒWn*tk rVtˆd ˆj¡dƒ‚YnXW5tˆd ˆj¡t|ƒƒX|S)Nz{}.duration.msz {}.successr5z {}.failure)Úpublish_metricr"rr?r*r@)rArBÚ start_timerE)Úfuncr/rrrGœs   ýz&profile..wrap..wrapperr)rLrGr2)rLrrH›szprofile..wrap)ÚwarningsÚwarnÚDeprecationWarning)r/rHrr2rÚprofileŽs  rP©r(r)Ú metric_groupcCst|ƒ ||¡dS)z• Publishes a metric data point. Usage :: put_metric("metric_name", 1) put_metric("metric_name", 1, "metric_group_name") N)r3r+rQrrrr>²s r>)rRr(r)cCs"t d¡t|ƒ}| ||¡dS)NzKDeprecated, use put_metric(metric_group)(metric_name, metric_value) instead)rMrNr3r+)rRr(r)Z metric_streamrrrrJÁs ÿrJ)Ústart_time_in_secondscCst ¡}t||dƒS)zG Returns the elapsed time in millis from the given start time. iè)r*r,)rSZend_timerrrr?Ésr?)N)Nr=)N)r=)rr*rMÚ collectionsrÚ functoolsrÚtypingrrrr ÚABCrr r$r%r0r-Ú__annotations__rr1r3r<rIrPr,r>rJÚfloatr?rrrrÚ s,       , $