U d@sddlZddlZddlZddlZddlZddlZddlmZddlm Z m Z ddl m Z m Z mZmZmZmZeeeefdddZeeedd d Zeeeeeefd d d ZeedddZeeeeeffddddZGdddZdS)N) timedelta)EventThread)AnyCallableDictOptionalTupleUnion) config_strreturncCsi}|}|s|S|d}|D]Z}|dd^}}|}|sJtd|r\|d}nd}|sttd|d|||<q"|S) zExtracts 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_valueskvkeyvaluesvaluerN/tmp/pip-unpacked-wheel-ua33x9lu/torch/distributed/elastic/rendezvous/utils.py_parse_rendezvous_configs*   r)port_strr cCs|rtd|rt|SdS)z3Tries to extract the port number from ``port_str``.z ^[0-9]{1,5}$N)rematchint)rrrr_try_parse_port8sr )endpoint default_portr cCs|dk r|}|sd|fS|ddkrD|ddkrD|fg^}}n|dd^}}t|dkr|ddkr|ddkr|dd}t|dkrt|d}|dks|d krtd |d n|}td |std |d||fS)a5Extracts 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"hostrestportrrrparse_rendezvous_endpoint?s* $      r-)r*r cCs|dkr dSzt|}Wntk r2d}YnX|rB|jrBdSt}||krVdStj|dtjtjd}|D]<}|dr|d|krdS|rp|ddt |krpdSqpdS) a=Indicates 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)protoflagsrF) ipaddress ip_addressr is_loopbacksocket gethostname getaddrinfo IPPROTO_TCP AI_CANONNAMEstr)r*addrZ this_hostZ addr_listZ addr_inforrr_matches_machine_hostnamens.  r<)secondsr cCs*t|trtj|}|dkr&t|dS)zSuspends 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)r=rrr_delays  rDc@seZdZUdZGdddZeeed<eeed<ee j ed<eed<e e de e d d d d Zeeed ddZed dddZd d ddZd d ddZed d ddZeddZd S)_PeriodicTimerzRepresents a timer that periodically runs a specified function. Args: interval: The interval, in seconds, between each run. function: The function to run. c@sJeZdZUeed<eded<eedfed<ee efed<e ed<dS) z_PeriodicTimer._Contextinterval.Nfunction.argskwargs stop_eventN) __name__ __module__ __qualname__float__annotations__rr rrr:rrrrr_Contexts  rQ_name_thread _finalizer_ctxrGN)rFrHrIrJr cOsVd|_||_||j_||j_|p*d|j_|p6i|j_t|j_ d|_ d|_ dS)Nr) rRrQrU total_secondsrFrHrIrJrrKrSrT)selfrFrHrIrJrrr__init__s     z_PeriodicTimer.__init__)r cCs|jS)zGets the name of the timer.)rRrWrrrnamesz_PeriodicTimer.name)rZr cCs|jrtd||_dS)zSets the name of the timer. The specified name will be assigned to the background thread and serves for debugging and troubleshooting purposes. The timer has already started.N)rS RuntimeErrorrR)rWrZrrrset_namesz_PeriodicTimer.set_namecCs\|jrtdt|j|jpd|jfdd|_t||j|j|jj |_ d|j _ |j dS)zStart the timer.r[Z PeriodicTimerT)targetrZrIdaemonFN) rSr\r_runrRrUweakreffinalize _stop_threadrKrTatexitstartrYrrrres z_PeriodicTimer.startcCs|jr|dS)z'Stop the timer at the next opportunity.N)rTrYrrrcancelsz_PeriodicTimer.cancelcCs$|j|js |j|j|jqdSN)rKwaitrFrHrIrJ)ctxrrrr`sz_PeriodicTimer._runcCs||dSrg)setjoin)threadrKrrrrcsz_PeriodicTimer._stop_thread)rLrMrN__doc__rQrr:rPrrarbrrrrXpropertyrZr]rerf staticmethodr`rcrrrrrEs*      rE)r2r@rr5rBradatetimer threadingrrtypingrrrrr r r:rrr r-boolr<rOrDrErrrrs  &/&