U Jºcèã @sLUddlZddlZddlZddlmZddlmZddlmZm Z ddddd d ddd ddddg Z edddddgƒZGdd„dƒZGdd„dej ƒZGdd„deƒZGdd„deƒZGdd „d ƒZiZeƒaeed<d+eedœdd „Zedœdd„Zd d!„Zd,edœd#d„Zd-d$d „Zd.eeed%œd&d„Zeeed'œd(d„Zed)œd*d„ZdS)/éN)Ú namedtuple©Úwraps)ÚDictÚOptionalÚ MetricsConfigÚ MetricHandlerÚConsoleMetricHandlerÚNullMetricHandlerÚMetricStreamÚ configureÚ getStreamÚprofÚprofileÚ put_metricÚpublish_metricÚget_elapsed_time_msÚ MetricDataÚ timestampÚ group_nameÚnameÚvaluec@s.eZdZdgZdeeeefdœdd„ZdS)rÚparamsN©rcCs||_|jdkri|_dS©Nr)Úselfr©rúI/tmp/pip-unpacked-wheel-gikjz4vx/torch/distributed/elastic/metrics/api.pyÚ__init__s zMetricsConfig.__init__)N)Ú__name__Ú __module__Ú__qualname__Ú __slots__rrÚstrrrrrrrsc@s eZdZejedœdd„ƒZdS)r©Úmetric_datacCsdSrr©rr%rrrÚemit!szMetricHandler.emitN)rr r!ÚabcÚabstractmethodrr'rrrrr sc@seZdZedœdd„ZdS)r r$cCs td |j|j|j|j¡ƒdS)Nz[{}][{}]: {}={})ÚprintÚformatrrrrr&rrrr''süÿzConsoleMetricHandler.emitN©rr r!rr'rrrrr &sc@seZdZedœdd„ZdS)r r$cCsdSrrr&rrrr'3szNullMetricHandler.emitNr,rrrrr 2sc@s,eZdZeedœdd„Zeedœdd„ZdS)r©rÚhandlercCs||_||_dSrr-)rrr.rrrr8szMetricStream.__init__)Úmetric_nameÚmetric_valuecCs |j tt ¡|j||ƒ¡dSr)r.r'rÚtimer)rr/r0rrrÚ add_value<sÿzMetricStream.add_valueN)rr r!r#rrÚintr2rrrrr7sÚ_default_metrics_handler©r.ÚgroupcCs|dkr|an|t|<dSr)r4Ú_metrics_mapr5rrrrGs©r6cCs |tkrt|}nt}t||ƒSr)r7r4r)r6r.rrrr Qs cCsR|j}| d¡}t|ƒdkrJ|j}|r@| d¡dd|dS|dSn|SdS)NÚ.ééÿÿÿÿr)r!ÚsplitÚlenr )ÚfnÚqualnamer<ÚmodulerrrÚ_get_metric_nameYs rAÚ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.successr:z.failure)rArrr1Ú Exception)ÚargsÚkwargsÚkeyÚstartÚresult)Úfr6rrÚwrapper|s z#prof..wrap..wrapperr)rIrJr8)rIrÚwrap{s zprof..wrapNr)r>r6rKrr8rrfscst 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)NcszzZz*t ¡}ˆ||Ž}tˆd ˆj¡dƒWn*tk rVtˆd ˆj¡dƒ‚YnXW5tˆd ˆj¡t|ƒƒX|S)Nz{}.duration.msz {}.successr:z {}.failure)rr+rrr1rC)rDrEÚ start_timerH)Úfuncr6rrrJ s ýz&profile..wrap..wrapperr)rMrJr8)rMrrKŸszprofile..wrap)ÚwarningsÚwarnÚDeprecationWarning)r6rKrr8rr’s©r/r0Ú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 r2rQrrrr¶s)rRr/r0cCs"t d¡t|ƒ}| ||¡dS)NzKDeprecated, use put_metric(metric_group)(metric_name, metric_value) instead)rNrOr r2)rRr/r0Z metric_streamrrrrÅs ÿ)Ústart_time_in_secondscCst ¡}t||dƒS)zG Returns the elapsed time in millis from the given start time. iè)r1r3)rSZend_timerrrrÍs)N)NrB)N)rB) r(r1rNÚcollectionsrÚ functoolsrÚtypingrrÚ__all__rrÚABCrr r rr7r4Ú__annotations__r#rr rArrr3rrÚfloatrrrrrÚ s@ þ , $