o eiw@s\UddlZddlZddlmZddlmZddlmZgdZedgdZ Gdd d Z Gd d d ej Z Gd d d e Z Gddde ZGdddZiZeee fed<eae ed<d.de dedBfddZdefddZddZd/defddZeded d.d!d"Zd0d#ed$ed%efd&d'Zed(ed d%ed#ed$efd)d*Zd+efd,d-ZdS)1N) namedtuplewraps) deprecated) MetricsConfig MetricHandlerConsoleMetricHandlerNullMetricHandler MetricStream configure getStreamprofprofile put_metricpublish_metricget_elapsed_time_ms MetricDatar) timestamp group_namenamevaluec@s.eZdZdgZddeeefdBfddZdS)rparamsNcCs||_|jdur i|_dSdSN)r)selfrro/var/www/addictedbytheproject.nl/epg/venv/lib/python3.10/site-packages/torch/distributed/elastic/metrics/api.py__init__'s  zMetricsConfig.__init__r)__name__ __module__ __qualname__ __slots__dictstrrrrrrr$s rc@s eZdZejdefddZdS)r metric_datacCdSrrrr#rrremit.szMetricHandler.emitN)rrrabcabstractmethodrr&rrrrr-src@eZdZdefddZdS)rr#c Cs,td|jd|jd|jd|jdS)N[z][z]: =)printrrrrr%rrrr&4s"zConsoleMetricHandler.emitNrrrrr&rrrrr3rc@r))r r#cCr$rrr%rrrr&;szNullMetricHandler.emitNr-rrrrr :r.r c@s0eZdZdedefddZdedefddZd S) r rhandlercCs||_||_dSr)rr/)rrr/rrrr@s zMetricStream.__init__ metric_name metric_valuecCs |jtt|j||dSr)r/r&rtimer)rr0r1rrr add_valueDszMetricStream.add_valueN)rrrr"rrintr3rrrrr ?sr _metrics_map_default_metrics_handlerr/groupcCs|dur|adS|t|<dSr)r6r5)r/r7rrrr Os r cCst|t}t||Sr)r5getr6r )r7r/rrrr Ys  r cCsL|j}|d}t|dkr$|j}|r |ddd|dS|dS|S)N.r)rsplitlenr)fnqualnamer<modulerrr_get_metric_name^s  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)rAr2r Exceptionr)argskwargskeystartresult)fr7rrwrappers z#prof..wrap..wrapperr)rKrLr7)rKrwraps zprof..wrapr)r>r7rNrrMrr ks r zDeprecated, 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)r2rrrEr)rFrG start_timerJ)funcr7rrrLs(   z&profile..wrap..wrapperr)rQrLrM)rQrrNszprofile..wrapr)r7rNrrMrrs rr0r1 metric_groupcCst|||dS)z Publish a metric data point. Usage :: put_metric("metric_name", 1) put_metric("metric_name", 1, "metric_group_name") Nr r3)r0r1rRrrrrs rzMDeprecated, use `put_metric(metric_group)(metric_name, metric_value)` insteadcCst|}|||dSrrS)rRr0r1 metric_streamrrrrsrstart_time_in_secondscCst}t||dS)zs6        , &