U yh@spddlZddlmZmZddlmZddlmZdgZdZ ee eddd d Z eeedd d dZ dS) N)contextmanager ExitStack) Generator)record worker_mainTORCH_WORKER_SERVER_SOCKET)NNN) socket_pathreturnccs.ddlm}||}z dVW5|XdS)Nr) _WorkerServer)Ztorch._C._distributed_c10dr shutdown)rr serverr Y/var/www/html/venv/lib/python3.8/site-packages/torch/distributed/elastic/control_plane.py_worker_servers   r)r c cs>t.}tjt}|dk r*|t|dVW5QRXdS)aw This is a context manager that wraps your main entry function. This combines the existing ``errors.record`` logic as well as a new ``_WorkerServer`` that exposes handlers via a unix socket specified by ``Torch_WORKER_SERVER_SOCKET``. Example :: @worker_main() def main(): pass if __name__=="__main__": main() N)rosenvirongetr enter_contextr)stackrr r rrs  ) r contextlibrrtypingrZ0torch.distributed.elastic.multiprocessing.errorsr__all__rstrrrr r r rs