i 3 UddlZddlZddlZddlZddlZddlZddlZddlmZm Z ddl m Z m Z ddl mZddlmZddlmZmZddlmZmZddlmZmZdd lmZmZej:eZej@e!d <ed Gd dZ"ed GddZ#ed GddZ$GddeZ%de&e'de e"fdZ(de&e'de e"fdZ)d/de'de'de*e&e'e e"ffdZ+de'fdZ,dZ-dZ.d Z/Gd!d"Z0Gd#d$eZ1Gd%d&eZ2Gd'd(Z3d)e4d*e'dzd+e5d,e&e%d-e6e'ddf d.Z7y)0N)ABCabstractmethod)CallableIterable)ThreadPoolExecutor) dataclass)BaseHTTPRequestHandlerThreadingHTTPServer)parse_qsurlparse) DictLoader Environment)get_world_sizetcpstore_clientloggerT)slotsc.eZdZUeed<eed<dZdZy)Response status_codetextcl|jdk7r%td|jd|jy)NzHTTP z: )r RuntimeErrorrselfs r/var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/torch/distributed/debug/_frontend.pyraise_for_statuszResponse.raise_for_status!s8   s "t'7'7&8499+FG G #c@tj|jSN)jsonloadsrrs rr!z Response.json%szz$))$$rN)__name__ __module__ __qualname__int__annotations__strrr!rrrrs IH%rrc"eZdZUeed<eed<y)NavLinkpathlabelN)r#r$r%r(r'r)rrr+r+)s I Jrr+c.eZdZUeed<edgefed<y)Router,HTTPRequestHandlerhandlerN)r#r$r%r(r'rbytesr)rrr/r//s I +,e3 44rr/cxeZdZedeefdZedeefdZde e e ffdZ de dzfdZ de fdZ y) DebugHandlerreturncyr r)rs rrouteszDebugHandler.routes6s%(rcyr r)rs r nav_linkszDebugHandler.nav_links9s*-rciSr r)rs r templateszDebugHandler.templates<s rNcyr r)rs rdumpzDebugHandler.dump?srcHt|jjSr )typer#lowerrs r dump_filenamezDebugHandler.dump_filenameBsDz""((**r)r#r$r%rlistr/r7r+r9dictr(r;r=rAr)rrr4r45s`(U ((-4=--4S>cDj+s+rr4urlsr5cddld}dtdtffd }t|5}|j ||}ddd|S#1swYSxYw)Nrurlr5cfj|}t|j|jSr )postrrr)rGresprequestss rgetzfetch_thread_pool..getQs(}}S!(($))44r) max_workers)rKr(rrmap)rDrMrLexecutorrespsrKs @rfetch_thread_poolrQKsXK555  4( S$'( L( Ls AA cddldjdtdtfddttdt tffd }t j||S)NrsessionrGr5cK|j|4d{}|jd{}t|j|cdddd{S7C7-7 #1d{7swYyxYwwr )rIrrstatus)rSrGrJrs rfetchzfetch_aiohttp..fetch`sc<<$ / /$DDKK. / / /$ / / / /sTA7AA7A"AA" A7A A7A" A7"A4(A+ )A40A7rDc Kj4d{}tj|Dcgc] }|| c}d{cdddd{S7Dcc}w77#1d{7swYyxYwwr ) ClientSessionasynciogather)rDrSrGaiohttprVs rrZzfetch_aiohttp..gatherest((* P Pg )N#%*=)NOO P P P)NO P P P PsaA=AA=A(A A(A$A( A=A&A=A(&A=(A:.A1 /A:6A=)r[rXr(rrBrrYrun)rDrZr[rVs @@r fetch_aiohttpr][s\/W22/// P49P();P ;;vd| $$rendpointargsc6t}ttDcgc]}d| }}|j|}|Dcgc]}|j d|d|}} t |}||fScc}wcc}w#t $rt|}Y||fSwxYw)Nrankz /handler/?)rranger multi_getdecoder] ImportErrorrQ)r^r_storerkeysaddrsaddrrPs r fetch_allrlls  E %n&6 7 81d1#J 8D 8 OOD !EFK Ld iz4& 9 LE L)e$ %< 9 L )!%( %<)s A4A9% A>>BBblobcZtj|}tj|dS)N)indent)r!r"dumps)rmparseds r format_jsonrszs! ZZ F ::fQ ''rac {% block title %}{% endblock %} - PyTorch Distributed
{% block header %}{% endblock %} {% block content %}{% endblock %}
a {% extends "base.html" %} {% block header %}

{% block title %}{{title}}{% endblock %}

{% endblock %} {% block content %} {% for i, (addr, resp) in enumerate(zip(addrs, resps)) %}

Rank {{ i }}: {{ addr }}

{% if resp.status_code != 200 %}

Failed to fetch: status={{ resp.status_code }}

{{ resp.text }}
{% else %} 
{{ resp.text }}
{% endif %} {% endfor %} {% endblock %} a {% extends "base.html" %} {% block header %}

{% block title %}{{ title }}{% endblock %}

{% endblock %} {% block content %} {% for i, (addr, resp) in enumerate(zip(addrs, resps)) %}

Rank {{ i }}: {{ addr }}

{% if resp.status_code != 200 %}

Failed to fetch: status={{ resp.status_code }}

{{ resp.text }}
{% else %} 
{{ format_json(resp.text) }}
{% endif %} {% endfor %} {% endblock %} c FeZdZ d deedededdfdZd dZd dZ d d Z y) PeriodicDumperhandlers output_dirinterval_secondsr5Ncn||_||_||_tj|_d|_yr ) _handlers _output_dir_interval_seconds threadingEvent _stop_event_thread)rrvrwrxs r__init__zPeriodicDumper.__init__s1 "%!1$??,04 rctj|jdtj|j dd|_|j jy)NT)exist_okz distributed.debug.PeriodicDumper)targetdaemonname)osmakedirsr{r}Thread_runrstartrs rrzPeriodicDumper.startsG D$$t4 ''993  rc|jj|j|jjyyr )rsetrjoinrs rstopzPeriodicDumper.stops3  << # LL    $rc|jjs|jD]} |j}|tjd}|jd|d}tjj|j|} t|d5}|j|ddd|jj!|j"|jjsyy#t$r(t j d|jYwxYw#1swYxYw#t$rt j d|Y9wxYw)NzFailed to dump %sz %Y%m%d_%H%M%S_z.txtwzFailed to write dump to %s)ris_setrzr= Exceptionr exceptionrAtimestrftimerr,rr{openwritewaitr|)rr1content timestampfilenamer,fs rrzPeriodicDumper._run s9""))+>> I%llnG? MM/: %3356a {$Gww||D$4$4h?IdC)A() I    ! !$"8"8 9#""))+!$$%8':O:O:QR)) I$$%A4HIs;C> D> D22D>>-D/.D/2D; 7D>>E! E!)gN@r5N) r#r$r%rBr4r(floatrrrrr)rrrurusF #' 5|$ 5 5 5  5 :rrucReZdZUejZej ed<dZe ed<y)_IPv6HTTPServeraddress_familyirequest_queue_sizeN) r#r$r%socketAF_INET6r AddressFamilyr'rr&r)rrrr#s!+1??NF((:""rrc veZdZUded<dZdZdefdZdeee effdZ defdZ d efd ed e d e de fd Zy )r0FrontendServerfrontendcRtjd|jd||zy)Nz%s %sr)rinfoclient_address)rformatr_s r log_messagezHTTPRequestHandler.log_message+s&     " TM rc:|jj|yr )r_handle_requestrs rdo_GETzHTTPRequestHandler.do_GET2s %%d+rr5c@t|jjSr )r r,rs rget_pathzHTTPRequestHandler.get_path5s "'''rc4t|jSr )r get_raw_queryrs r get_queryzHTTPRequestHandler.get_query8s**,--rc@t|jjSr )r r,queryrs rrz HTTPRequestHandler.get_raw_query;s "(((rNrdefaultr?cJ|j}||vr|S|||dS)Nr)r)rrrr?rs r get_query_argz HTTPRequestHandler.get_query_arg>s0  u NE$KN##r)r#r$r%r'rrr(rrCrBrrobjectr?rr)rrr0r0(so ,(#(.4T#Y/.)s),0c$$"($7;$ $rr0c`eZdZ d dedeedzfdZddZddZde ddfd Z d e d e de fd Zy)rNportrvc| ddlm}|}djd|D}ttt d}|D]!}|j |j#t|}t|d|_ |jjj ttt|i|_|D]:}|j!D]%}|j"|j|j$<'<||_t)d t*fd |i} d |f} t-| | |_t1j2|j4d dd |_|j6j9y)Nr)default_handlers c3K|]6}|jD]!}d|jd|jd#8yw)z z N)r9r,r-).0r1links r z*FrontendServer.__init__..SsN ))+ DII;b 4L M M s<>)z base.htmlz raw_resp.htmlzjson_resp.htmlT)loader enable_async)ziprs enumerater9r0rr)z distributed.debug.FrontendServer)rr_rr)'torch.distributed.debug._debug_handlersrr BASE_TEMPLATERAW_RESP_TEMPLATEJSON_RESP_TEMPLATEupdater;r r _jinja_envglobalsrrsr_routesr7r1r,rzr?r0r_serverr}r_serverr) rrrvrnav_html all_templatesr1rrouteRequestHandlerClassserver_addresss rrzFrontendServer.__init__Hsj   P')H99 #  '.0)   6G  !2!2!4 5 6M*%V$G &&# ' JL  9G ) 9+0== UZZ( 9 9"#  !    d&~7JK  '';;3   rr5c |jjy#t$rtj dYywxYw)Nz got exception in frontend server)r serve_foreverrrrrs rrzFrontendServer._serves8 A LL & & ( A   ? @ As >>c8|jjyr )rrrs rrzFrontendServer.joins rreqc|j}||jvr|jdd|y|j|} ||}|jd|jdd|j|jj|y#ttf$r>}t j d||jddt|Yd}~yd}~wwxYw) NizHandler not found: z-Exception in frontend server when handling %siz Exception: rz Content-typez text/html) rr send_errorr SystemExitrrrepr send_response send_header end_headerswfiler)rrr,r1rJes rrzFrontendServer._handle_requests||~ t|| # NN3"5dV < = ,,t$ 3rs   #.1!C+*J+**844  % % %   555 +3+, DI (8*<  %S %hx&8%"  3 d3i(AS6S0T (c(> @$0-:-:j#)# $/$>_P_PD" "Dj""< " s8 "  "r