o h}@s@UddlZddlZddlZddlmZddlmZddlmZm Z gdZ edgdZ Gdd d Z Gd d d ej ZGd d d eZGdddeZGdddZiZeaeed<d*dede efddZdefddZddZd+defddZd*ddZd,d ed!ed"efd#d$Zd"ed ed!efd%d&Zd'efd(d)ZdS)-N) namedtuplewraps)DictOptional) MetricsConfig MetricHandlerConsoleMetricHandlerNullMetricHandler MetricStream configure getStreamprofprofile put_metricpublish_metricget_elapsed_time_ms MetricDatar) timestamp group_namenamevaluec@s.eZdZdgZddeeeeffddZdS)rparamsNcCs||_|jdur i|_dSdSN)r)selfrr[/var/www/html/ai/venv/lib/python3.10/site-packages/torch/distributed/elastic/metrics/api.py__init__s  zMetricsConfig.__init__r)__name__ __module__ __qualname__ __slots__rrstrrrrrrrs rc@s eZdZejdefddZdS)r metric_datacCdSrrrr#rrremit!szMetricHandler.emitN)rrr abcabstractmethodrr&rrrrr src@eZdZdefddZdS)r r#c Cs,td|jd|jd|jd|jdS)N[z][z]: =)printrrrrr%rrrr&'s"zConsoleMetricHandler.emitNrrr rr&rrrrr &r c@r))r r#cCr$rrr%rrrr&.szNullMetricHandler.emitNr-rrrrr -r.r c@s0eZdZdedefddZdedefddZd S) r rhandlercCs||_||_dSr)rr/)rrr/rrrr3s zMetricStream.__init__ metric_name metric_valuecCs |jtt|j||dSr)r/r&rtimer)rr0r1rrr add_value7szMetricStream.add_valueN)rrr r"rrintr3rrrrr 2sr _default_metrics_handlerr/groupcCs|dur|adS|t|<dSr)r5 _metrics_map)r/r6rrrr Bs r cCs |tvr t|}nt}t||Sr)r7r5r )r6r/rrrr Ls  r cCsL|j}|d}t|dkr$|j}|r |ddd|dS|dS|S)N.r)r splitlenr)fnqualnamer;modulerrr_get_metric_nameTs  r@ torchelasticcsfdd}|r ||S|S)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 ctfdd}|S)Nc st}z6zt}|i|}t|ddWnty,t|ddwWt|dt||St|dt|wNz.successr9z.failurez .duration.ms)r@r2r Exceptionr)argskwargskeystartresult)fr6rrwrapperws z#prof..wrap..wrapperr)rJrKr6)rJrwrapvs zprof..wrapr)r=r6rMrrLrras 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 insteadcrB)Nc sz9zt}|i|}tjddWnty*tjddwWtjdt||Stjdt|wrC)r2rrrDr)rErF start_timerI)funcr6rrrKs(   z&profile..wrap..wrapperr)rOrKrL)rOrrMszprofile..wrap)warningswarnDeprecationWarning)r6rMrrLrrs rr0r1 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 r3)r0r1rSrrrrs rcCs"tdt|}|||dS)NzKDeprecated, use put_metric(metric_group)(metric_name, metric_value) instead)rPrQr r3)rSr0r1 metric_streamrrrrs rstart_time_in_secondscCst}t||dS)zG Returns the elapsed time in millis from the given start time. i)r2r4)rUend_timerrrrsrr)NrA)rA) r'r2rP collectionsr functoolsrtypingrr__all__rrABCrr r r r7r5__annotations__r"r r r@rrr4rrfloatrrrrrs.       ,$