VhddlZddlmZmZddlmZmZddlZ ddl Z ddl mZddlmZmZmZddlmZmZdd lmZdd lmZGd d eZd ede j6fdZd edeeeffdZy#e $r ddl m Z Y\wxYw)N) b64decode b64encode)castOptional) _etcd_stub)Store)RendezvousConnectionErrorRendezvousParametersRendezvousStateError)RendezvousBackendToken) EtcdStore)parse_rendezvous_endpointc eZdZUdZdZej ed<eed<e ed< ddej ded e ed e e d df d Z e d efd Z d e eeeffdZ ddede ed e eeeeffdZdej(d eeeffdZy)EtcdRendezvousBackendaRepresents an etcd-based rendezvous backend. Args: client: The ``etcd.Client`` instance to use to communicate with etcd. run_id: The run id of the rendezvous. key_prefix: The path under which to store the rendezvous state in etcd. ttl: The TTL of the rendezvous state. If not specified, defaults to two hours. i _client_key_ttlNclientrun_id key_prefixttlreturnc|s td||_|r|dz|z|_n||_|r |dkDr||_y|j|_y)Nz&The run id must be a non-empty string./r) ValueErrorrrr _DEFAULT_TTL)selfrrrrs l/home/dcms/DCMS/lib/python3.12/site-packages/torch/distributed/elastic/rendezvous/etcd_rendezvous_backend.py__init__zEtcdRendezvousBackend.__init__0sREF F "S(61DIDI 37DI))DIcy)See base class.zetcd-v2)rs r namezEtcdRendezvousBackend.nameFsr"c |jj|j}|j|S#tj$rYytj t jjf$r}td|d}~wwxYw)r$NCThe connection to etcd has failed. See inner exception for details.) rreadretcdEtcdKeyNotFound EtcdExceptionurllib3 exceptions TimeoutErrorr _decode_state)rresultexcs r get_statezEtcdRendezvousBackend.get_stateKs \\&&tyy1F!!&))## ""G$6$6$C$CD +U  s%8B  ,B 9 BB statetokenct|j}i}fd}|r t|}|r||d<nd|d< jj j |jfi|}||Sgj!|d}|S#t$r |cYSwxYw#tjtjf$rd}Y^tjtjjf$r}td|d}~wwxYw)r$c<j}| g|d}|Sy)NF)r3)r1tmprs r r3z2EtcdRendezvousBackend.set_state..get_state`s,^^%F!$vnun r" prevIndexF prevExistNr(T)rdecodeintrrwriterrr*EtcdAlreadyExistEtcdCompareFailedr,r-r.r/r r0) rr4r5 base64_statekwargsr3r1r2r8s ` r set_statezEtcdRendezvousBackend.set_stateXs!'..0    #E  "'F; "'F;  'T\\'' <UfUF >; /t!!&)/4/ + # {" #%%t'='=> F""G$6$6$C$CD +U  s. B3B(B%$B%(%D ,D ; DD r1c|jj} t|}||j fS#tj$r}t d|d}~wwxYw)Nz=The state object is corrupt. See inner exception for details.)valueencoderbinasciiErrorr modifiedIndex)rr1r@r4r2s r r0z#EtcdRendezvousBackend._decode_states_||**,  l+E f**** ~~ &O  s 5A AA)NN)N)__name__ __module__ __qualname____doc__rr*Client__annotations__strr<rr!propertyr&tuplebytesrr3boolrB EtcdResultr0r%r"r rrs L [[ I I %)! * **SM * c] *  *,c *8E%,$78 *6:))#+E?) %ud*+ ,)V +DOO +eUl8K +r"rparamsrc Nt|jd\}}tt|j dd}|dkr t d|j ddjj}|dk7r|d k7r t d |j d }|r|j d }|r||f}|j d } tj||||||dS#tjtjjf$r}td|d}~wwxYw)NiK ) default_port read_timeout<rz,The read timeout must be a positive integer.protocolhttphttpsz#The protocol must be HTTP or HTTPS.ssl_cert ssl_cert_keyca_certT)rXrZcertr_allow_reconnectr()rendpointrr< get_as_intrgetstriplowerr*rMr,r-r.r/r ) rUhostportrXrZr]r^r_r2s r _create_etcd_clientris+*6??NJD$V..~rBCLqGHHzz*f-335;;=H6h'1>??zz*%Hzz.1 !,/Hjj#G {{  %       2 2 ? ? @' Q  s C&&-D$ DD$cht|}t||jd}t|d}||fS)aCreate a new :py:class:`EtcdRendezvousBackend` from the specified parameters. +--------------+-----------------------------------------------------------+ | Parameter | Description | +==============+===========================================================+ | read_timeout | The read timeout, in seconds, for etcd operations. | | | Defaults to 60 seconds. | +--------------+-----------------------------------------------------------+ | protocol | The protocol to use to communicate with etcd. Valid | | | values are "http" and "https". Defaults to "http". | +--------------+-----------------------------------------------------------+ | ssl_cert | The path to the SSL client certificate to use along with | | | HTTPS. Defaults to ``None``. | +--------------+-----------------------------------------------------------+ | ssl_cert_key | The path to the private key of the SSL client certificate | | | to use along with HTTPS. Defaults to ``None``. | +--------------+-----------------------------------------------------------+ | ca_cert | The path to the rool SSL authority certificate. Defaults | | | to ``None``. | +--------------+-----------------------------------------------------------+ z/torch/elastic/rendezvous)rz/torch/elastic/store)rirrr)rUrbackendstores r create_backendrms=,! (F# *EG f4 5E E>r")rFbase64rrtypingrrurllib3.exceptionsr-r*ModuleNotFoundErrorrtorch.distributedr apir r r dynamic_rendezvousr r etcd_storerutilsrrrMrirQrmr%r"r rxs'!%$VV8!,q+-q+h& 4&&R/E:OQV:V4WQ%$%sA-- A;:A;