o H&i!@s ddlZddlZddlZddlZddlZddlZddlmZddlm Z m Z ddl m Z m Z mZmZmZmZdgZdedeeeffdd Zd edeefd d Zd eededeeeffddZdedefddZdeeeeeffddfddZGdddZdS)N) timedelta)EventThread)AnyCallableDictOptionalTupleUnionparse_rendezvous_endpoint config_strreturncCsi}|}|s |S|d}|D]-}|dd^}}|}|s%td|r.|d}nd}|s:td|d|||<q|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_valueskvkeyvaluesvaluerUC:\wamp64\www\opt\env\Lib\site-packages\torch/distributed/elastic/rendezvous/utils.py_parse_rendezvous_configs*   rport_strcCs|r td|r t|SdS)z1Try to extract the port number from ``port_str``.z ^[0-9]{1,5}$N)rematchint)rrrr_try_parse_port9sr!endpoint default_portcCs|dur|}|sd|fS|ddkr#|ddkr#|ggR^}}n|dd^}}t|dkrC|ddkrC|ddkrC|dd}t|dkr`t|d}|dusW|d kr_td |d n|}td |sptd |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#hostrestportrrrr @s, $      r+c Cs|dkrdSzt|}Wn tyd}Ynw|r |jr dSz tj|dtjtjd}WnttjfyD}zg}WYd}~nd}~wwdd|D}t }||krVdStj|dtjtjd}|D])}|drr|d|krrdS|r|dd t |krdS|dd |vrdSqcd 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)protoflagscSsg|]}|ddqS)rr).0Zhost_addr_inforrr s z-_matches_machine_hostname..r0rF) ipaddress ip_addressr is_loopbacksocket getaddrinfo IPPROTO_TCP AI_CANONNAMEgaierror gethostnamestr)r+addrZhost_addr_list_Z host_ip_listZ this_hostZ addr_list addr_inforrr_matches_machine_hostnameosF     rAsecondscCs.t|tr tj|}|dkrt|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)rBrrr_delays  rIc @seZdZUdZGdddZeeed<eeed<ee j ed<eed<de d e d d e d e d df ddZed eefddZded dfddZdddZdddZedddZeddZdS)_PeriodicTimerzRepresent 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   rV_name_thread _finalizer_ctxrKrMrLrNrOr NcOsVd|_||_||j_||j_|pd|j_|pi|j_t|j_ d|_ d|_ dS)Nr) rWrVrZ total_secondsrKrMrNrOrrPrXrY)selfrKrMrNrOrrr__init__s      z_PeriodicTimer.__init__cCs|jS)zGet the name of the timer.)rWr\rrrnamesz_PeriodicTimer.namer_cCs|jrtd||_dS)zSet 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)rX RuntimeErrorrW)r\r_rrrset_names z_PeriodicTimer.set_namecCs\|jrtdt|j|jpd|jfdd|_t||j|j|jj |_ d|j _ |j dS)zStart the timer.r`Z PeriodicTimerT)targetr_rNdaemonFN) rXrar_runrWrZweakreffinalize _stop_threadrPrYatexitstartr^rrrrjsz_PeriodicTimer.startcCs|jr |dSdS)z'Stop the timer at the next opportunity.N)rYr^rrrcancel s z_PeriodicTimer.cancelcCs8|j|js|j|ji|j|j|jrdSdSN)rPwaitrKrMrNrO)ctxrrrresz_PeriodicTimer._runcCs||dSrl)setjoin)threadrPrrrrhs z_PeriodicTimer._stop_thread)r N)rQrRrS__doc__rVrr=rUrrfrgrrrr]propertyr_rbrjrk staticmethodrerhrrrrrJs6      rJ)r4rErr7rGrfdatetimer threadingrrtypingrrrrr r __all__r=rr r!r boolrArTrIrJrrrrs   &"/"7