sgddlZddlmZmZddlmZddlmZdgZdZ ede ded fd Z eeded fd Z y) N)contextmanager ExitStack) Generator)record worker_mainTORCH_WORKER_SERVER_SOCKET socket_pathreturn)NNNc#|Kddlm}||} d|jy#|jwxYww)Nr) _WorkerServer)torch._C._distributed_c10dr shutdown)r r servers Z/var/www/html/venv/lib/python3.12/site-packages/torch/distributed/elastic/control_plane.py_worker_serverrs/8 ; 'F s<'<9<c#Kt5}tjjt}||j t |ddddy#1swYyxYww)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)stackr s rrrsR* jjnn%?@  "   { ; < s A&AA A&A#A&) r contextlibrrtypingr0torch.distributed.elastic.multiprocessing.errorsr__all__rstrrrrrsv 0C : 2B(CY/0r