Wj  ddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z ddl mZdgZdedeeeffd Zd ededzfd Zd edzd edeeeffdZdedefdZdeeeefzddfdZGddZdS)N)Callable) timedelta)EventThread)Anyparse_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 o/home/longshao/multi-rider-rag/.venv/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 rrrDsT>>## +\**{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(NNNrr2rF) 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_hostnamerIus {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)rJs r_delayrRsH'5!!+.'*$ 7rc eZdZUdZGddZedzed<edzed<ej dzed<eed<d e d e d d e d e ddf dZ ededzfdZdeddfdZddZddZeddZedZdS)_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__rrMrdictrArr3rr_ContextrVs`9%%%%CHoS#XrrcN_name_thread _finalizer_ctxrWrYrXrZr[r cd|_||_||j_||j_|pd|j_|pi|j_t|j_ d|_ d|_ dS)Nr3) rdrcrg total_secondsrWrYrZr[rr\rerf)selfrWrYrZr[s r__init__z_PeriodicTimer.__init__su MMOO %3355 %  !rs $$$$$$######## ' '##c3h####LccDj.Dj.(+. 38_....b1C1D1111h EE%,// D    eeeeeeeeeer