U ‰dªã@sŠddlmZmZddlmZddlmZeedœdd„Zeedœdd„Zeedœd d „Z eedœd d „Z d dœdd„Z eedœdd„Z d S)é)ÚRendezvousHandlerÚRendezvousParameters)Úrendezvous_handler_registry)Úcreate_handler)ÚparamsÚreturncCsddlm}| |¡S)Nr)Ústatic_tcp_rendezvous)ÚrÚcreate_rdzv_handler)rr©r úQ/tmp/pip-unpacked-wheel-ua33x9lu/torch/distributed/elastic/rendezvous/registry.pyÚ_create_static_handler s r cCsddlm}| |¡S)Nr)Úetcd_rendezvous)r rr )rrr r r Ú_create_etcd_handlers rcCs$ddlm}||ƒ\}}t|||ƒS©Nr)Úcreate_backend)Zetcd_rendezvous_backendrr©rrÚbackendÚstorer r r Ú_create_etcd_v2_handlers  rcCs$ddlm}||ƒ\}}t|||ƒSr)Zc10d_rendezvous_backendrrrr r r Ú_create_c10d_handler s  rN)rcCs4t dt¡t dt¡t dt¡t dt¡dS)NZetcdzetcd-v2Zc10dZstatic)Úhandler_registryÚregisterrrrr r r r r Ú_register_default_handlers(s   rcCs t |¡S)aL This method is used to obtain a reference to a :py:class`RendezvousHandler`. Custom rendezvous handlers can be registered by :: from torch.distributed.elastid.rendezvous import rendezvous_handler_registry from torch.distributed.elastic.rendezvous.registry import get_rendezvous_handler def create_my_rdzv(params: RendezvousParameters): return MyCustomRdzv(params) rendezvous_handler_registry.register("my_rdzv_backend_name", create_my_rdzv) my_rdzv_handler = get_rendezvous_handler("my_rdzv_backend_name", RendezvousParameters) )rr)rr r r Úget_rendezvous_handler/sr) ÚapirrrrZdynamic_rendezvousrr rrrrrr r r r Ús