`iddlZddlmZddlmZmZddlmZdgZdZ ede ded fd Z eeded fd Z dS) N) Generator)contextmanager ExitStack)record worker_mainTORCH_WORKER_SERVER_SOCKET socket_pathreturn)NNNc#Kddlm}||} dV|dS#|wxYw)Nr) _WorkerServer)torch._C._distributed_c10dr shutdown)r r servers {/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torch/distributed/elastic/control_plane.py_worker_serverrsa888888 ]; ' 'F s /Ac#Kt5}tjt}|"|t |dVddddS#1swxYwYdS)a{ 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)stackr s rrrs, jnn%?@@  "   { ; ; < < <  sA A++A/2A/) rcollections.abcr contextlibrr0torch.distributed.elastic.multiprocessing.errorsr__all__rstrrrrrs %%%%%%00000000CCCCCC : 2B(CY/0r