`i 6ddlZddlZddlZddlZddlZddlZddlmZddlm Z m Z ddl m Z m Z mZmZdgZdedeeeffdZd edeefd Zd eed edeeeffd ZdedefdZdeeeeeffddfdZGddZdS)N) timedelta)EventThread)AnyCallableOptionalUnionparse_rendezvous_endpoint config_strreturncZi}|}|s|S|d}|D]x}|dd^}}|}|std|r|d}nd}|std|d|||<y|S) zExtract key-value pairs from a rendezvous configuration string. Args: config_str: A string in format =,...,=. ,=zZThe rendezvous configuration string must be in format =,...,=.rNz%The rendezvous configuration option 'z' must have a value specified.)stripsplit ValueError)r config key_valueskvkeyvaluesvalues ~/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torch/distributed/elastic/rendezvous/utils.py_parse_rendezvous_configrs F!!##J  !!#&&JxxQ'' fiikk 7   1IOO%%EEE [[[[ s Mport_strcR|r$tjd|rt|SdS)z1Try to extract the port number from ``port_str``.z ^[0-9]{1,5}$N)rematchint)rs r_try_parse_portr"<s.BH_h778}} 4rendpoint default_portc||}|sd|fS|ddkr|ddkr |ggR^}}n|dd^}}t|dkr"|ddkr|ddkr |dd}t|dkr1t|d}||d krt d |d n|}t jd |st d |d||fS)a4Extract the hostname and the port number from a rendezvous endpoint. Args: endpoint: A string in format [:]. default_port: The port number to use if the endpoint does not include one. Returns: A tuple of hostname and port number. N localhostr[]:riz,The port number of the rendezvous endpoint 'z)' must be an integer between 0 and 65536.z ^[\w\.:-]+$z)The hostname of the rendezvous endpoint 'zN' must be a dot-separated list of labels, an IPv4 address, or an IPv6 address.)rrsplitlenr"rrr )r#r$hostrestports rr r CsT>>## +\**{chrlc11mmm ttooc1-- t 4yy1}}aCDHOOAbDz 4yyA~~tAw'' <45=='x''' )  8ND ) )  ; ; ; ;   :rr-cn|dkrdS tj|}n#t$rd}YnwxYw|r |jrdS t j|dtjtj}n%#ttjf$r }g}Yd}~nd}~wwxYwd|D}t j }||krdSt j|dtjtj}|D]P}|dr|d|krdS|r"|ddt|krdS|dd|vrdSQd S) a<Indicate whether ``host`` matches the hostname of this machine. This function compares ``host`` to the hostname as well as to the IP addresses of this machine. Note that it may return a false negative if this machine has CNAME records beyond its FQDN or IP addresses assigned to secondary NICs. r&TN)protoflagsc*g|]}|ddS)r).0host_addr_infos r z-_matches_machine_hostname..s"NNN^N1%a(NNNrr4rF) ipaddress ip_addressr is_loopbacksocket getaddrinfo IPPROTO_TCP AI_CANONNAMEgaierror gethostnamestr)r-addrhost_addr_list_ host_ip_list this_host addr_list addr_infos r_matches_machine_hostnamerKts {t#D))   t+ $f08K     (ON~NNNL"$$I yt"4v19LI   Q< IaLD0044  IaLOs4yy0044 Q<?l * *44 + 5s# ..,A**B BB secondsct|trtj|}|dkrt j|dSdS)zSuspend the current thread for ``seconds``. Args: seconds: Either the delay, in seconds, or a tuple of a lower and an upper bound within which a random delay will be picked. g{Gz?N) isinstancetuplerandomuniformtimesleep)rLs r_delayrTsH'5!!+.'*$ 7rc "eZdZUdZGddZeeed<eeed<ee j ed<eed<de d e d d e d e d df dZed eefdZded dfdZddZddZeddZedZdS)_PeriodicTimerzRepresent a timer that periodically runs a specified function. Args: interval: The interval, in seconds, between each run. function: The function to run. cneZdZUeed<eded<eedfed<ee efed<e ed<dS) _PeriodicTimer._Contextinterval.Nfunction.argskwargs stop_eventN) __name__ __module__ __qualname__float__annotations__rrOrdictrCrr5rr_ContextrXs`9%%%%CHoS#Xrre_name_thread _finalizer_ctxrYr[rZr\r]r Ncd|_||_||j_||j_|pd|j_|pi|j_t|j_ d|_ d|_ dS)Nr5) rfreri total_secondsrYr[r\r]rr^rgrh)selfrYr[r\r]s r__init__z_PeriodicTimer.__init__su MMOO %3355 %  !rs ########111111111111 ' '##c3h####Lchsm.sm.+.. 38_....b1C1D1111h E%ue|!445 $    eeeeeeeeeer