U ʹj @sddlZddlZddlZddlZddlZddlZddlmZddlm Z m Z ddl m Z m Z mZmZmZmZdgZeeeefdddZeeed d d Zeeeeeefd d dZeedddZeeeeeffddddZGdddZdS)N) timedelta)EventThread)AnyCallableDictOptionalTupleUnionparse_rendezvous_endpoint) 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_valueskvkeyvaluesvalueri/var/www/html/bwave/python/venv/lib/python3.8/site-packages/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_port9sr!)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#hostrestportrrrr @s* $      )r+r c Cs"|dkr dSzt|}Wntk r2d}YnX|rB|jrBdSztj|dtjtjd}Wn,ttjfk r}zg}W5d}~XYnXdd|D}t }||krdStj|dtjtjd}|D]V}|dr|d|krdS|r|dd t |krdS|dd |krdSqd S) 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)protoflagscSsg|]}|ddqS)rr).0Zhost_addr_inforrr sz-_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_listZ addr_inforrr_matches_machine_hostnameosJ   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)rArrr_delays  rHc@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  rU_name_thread _finalizer_ctxrKN)rJrLrMrNr cOsVd|_||_||j_||j_|p*d|j_|p6i|j_t|j_ d|_ d|_ dS)Nr) rVrUrY total_secondsrJrLrMrNrrOrWrX)selfrJrLrMrNrrr__init__s     z_PeriodicTimer.__init__)r cCs|jS)zGets the name of the timer.)rVr[rrrnamesz_PeriodicTimer.name)r^r 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)rW RuntimeErrorrV)r[r^rrrset_namesz_PeriodicTimer.set_namecCs\|jrtdt|j|jpd|jfdd|_t||j|j|jj |_ d|j _ |j dS)zStart the timer.r_Z PeriodicTimerT)targetr^rMdaemonFN) rWr`r_runrVrYweakreffinalize _stop_threadrOrXatexitstartr]rrrris z_PeriodicTimer.startcCs|jr|dS)z'Stop the timer at the next opportunity.N)rXr]rrrcancel sz_PeriodicTimer.cancelcCs$|j|js |j|j|jqdSN)rOwaitrJrLrMrN)ctxrrrrdsz_PeriodicTimer._runcCs||dSrk)setjoin)threadrOrrrrgsz_PeriodicTimer._stop_thread)rPrQrR__doc__rUrr=rTrrerfrrrr\propertyr^rarirj staticmethodrdrgrrrrrIs*      rI)r4rDrr7rFredatetimer threadingrrtypingrrrrr r __all__r=rr r!r boolr@rSrHrIrrrrs  &/7