o Zh,@shUddlZddlZddlmZddlmZddlmZddlm Z gdZ edgdZ Gd d d Z Gd d d ej ZGd ddeZGdddeZGdddZiZeeefed<eaeed<d/dedeefddZdefddZddZd0defddZe d ed!d/d"d#Zd1d$ed%ed&efd'd(Ze d)ed!d&ed$ed%efd*d+Zd,e fd-d.Z!dS)2N) namedtuplewraps)Optional) deprecated) MetricsConfig MetricHandlerConsoleMetricHandlerNullMetricHandler MetricStream configure getStreamprofprofile put_metricpublish_metricget_elapsed_time_ms MetricDatar) timestamp group_namenamevaluec@s.eZdZdgZddeeeeffddZdS)rparamsNcCs||_|jdur i|_dSdSN)r)selfrrT/var/www/auris/lib/python3.10/site-packages/torch/distributed/elastic/metrics/api.py__init__(s  zMetricsConfig.__init__r)__name__ __module__ __qualname__ __slots__rdictstrrrrrrr%s 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'5s"zConsoleMetricHandler.emitNrrr rr'rrrrr 4r c@r*)r r$cCr%rrr&rrrr'<szNullMetricHandler.emitNr.rrrrr ;r/r c@s0eZdZdedefddZdedefddZd S) r rhandlercCs||_||_dSr)rr0)rrr0rrrrAs zMetricStream.__init__ metric_name metric_valuecCs |jtt|j||dSr)r0r'rtimer)rr1r2rrr add_valueEszMetricStream.add_valueN)rrr r#rrintr4rrrrr @sr _metrics_map_default_metrics_handlerr0groupcCs|dur|adS|t|<dSr)r7r6)r0r8rrrr Ps r cCs |tvr t|}nt}t||Sr)r6r7r )r8r0rrrr Zs  r cCsL|j}|d}t|dkr$|j}|r |ddd|dS|dS|S)N.r)r splitlenr)fnqualnamer<modulerrr_get_metric_namebs  rA 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.successr:z.failurez .duration.ms)rAr3r Exceptionr)argskwargskeystartresult)fr8rrwrappers z#prof..wrap..wrapperr)rKrLr8)rKrwraps zprof..wrapr)r>r8rNrrMrros rzDeprecated, use `@prof` instead)categorycsfdd}|S)z @profile decorator adds latency and success/failure metrics to any given function. Usage :: @metrics.profile("my_metric_group") def some_function(): crC)Nc sz9zt}|i|}tjddWnty*tjddwWtjdt||Stjdt|wrD)r3rrrEr)rFrG start_timerJ)funcr8rrrLs(   z&profile..wrap..wrapperr)rQrLrM)rQrrNszprofile..wrapr)r8rNrrMrrs rr1r2 metric_groupcCst|||dS)z Publish a metric data point. Usage :: put_metric("metric_name", 1) put_metric("metric_name", 1, "metric_group_name") Nr r4)r1r2rRrrrrs rzMDeprecated, use `put_metric(metric_group)(metric_name, metric_value)` insteadcCst|}|||dSrrS)rRr1r2Z metric_streamrrrrsrstart_time_in_secondscCst}t||dS)zs8         , #