a d@s2UddlZddlZddlZddlmZddlmZddlmZm Z gdZ edgdZ Gdd d Z Gd d d ej ZGd d d eZGdddeZGdddZiZeaeed<d)eedddZedddZddZd*edddZd+ddZd,eeed d!d"Zeeed#d$d%Zed&d'd(ZdS)-N) namedtuplewraps)DictOptional) MetricsConfig MetricHandlerConsoleMetricHandlerNullMetricHandler MetricStream configure getStreamprofprofile put_metricpublish_metricget_elapsed_time_ms MetricDatar) timestamp group_namenamevaluec@s.eZdZdgZdeeeefdddZdS)rparamsNrcCs||_|jduri|_dSNr)selfrrn/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torch/distributed/elastic/metrics/api.py__init__s zMetricsConfig.__init__)N)__name__ __module__ __qualname__ __slots__rrstrrrrrrrsrc@s eZdZejedddZdS)r metric_datacCsdSrrrr%rrremit!szMetricHandler.emitN)rr r!abcabstractmethodrr'rrrrr src@seZdZedddZdS)r r$cCs td|j|j|j|jdS)Nz[{}][{}]: {}={})printformatrrrrr&rrrr''szConsoleMetricHandler.emitNrr r!rr'rrrrr &sr c@seZdZedddZdS)r r$cCsdSrrr&rrrr'3szNullMetricHandler.emitNr,rrrrr 2sr c@s,eZdZeedddZeedddZdS)r rhandlercCs||_||_dSrr-)rrr.rrrr8szMetricStream.__init__) metric_name metric_valuecCs |jtt|j||dSr)r.r'rtimer)rr/r0rrr add_value<szMetricStream.add_valueN)rr r!r#rrintr2rrrrr 7sr _default_metrics_handlerr.groupcCs|dur|an|t|<dSr)r4 _metrics_mapr5rrrr Gsr r6cCs |tvrt|}nt}t||Sr)r7r4r )r6r.rrrr Qs r cCsR|j}|d}t|dkrJ|j}|r@|ddd|dS|dSn|SdS)N.r)r!splitlenr )fnqualnamer<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 cstfdd}|S)Nc st}znz,t}|i|}t|ddWn&ty\t|ddYn0Wt|dt|nt|dt|0|S)Nz.successr:z.failurez .duration.ms)rAr1r Exceptionr)argskwargskeystartresult)fr6rrwrapper|s  0z#prof..wrap..wrapperr)rIrJr8)rIrwrap{s zprof..wrapNr)r>r6rKrr8rrfs rcstdtfdd}|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 sztz.t}|i|}tdjdWn(tyXtdjdYn0Wtdjt|ntdjt|0|S)Nz {}.successr:z {}.failurez{}.duration.ms)r1rr+rrCr)rDrE start_timerH)funcr6rrrJs"    z&profile..wrap..wrapperr)rMrJr8)rMrrKszprofile..wrap)warningswarnDeprecationWarning)r6rKrr8rrs  rr/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 r2rQrrrrs r)rRr/r0cCs"tdt|}|||dS)NzKDeprecated, use put_metric(metric_group)(metric_name, metric_value) instead)rNrOr r2)rRr/r0Z metric_streamrrrrs r)start_time_in_secondscCst}t||dS)zG Returns the elapsed time in millis from the given start time. i)r1r3)rSZend_timerrrrsr)N)NrB)N)rB) r(r1rN collectionsr functoolsrtypingrr__all__rrABCrr r r r7r4__annotations__r#r r rArrr3rrfloatrrrrr s,        , $