o Zh@srddlZddlmZddlmZmZddlmZdgZdZ ede ded fd d Z eeded fd dZ dS) N) Generator)contextmanager ExitStack)record worker_mainTORCH_WORKER_SERVER_SOCKET socket_pathreturn)NNNccs6ddlm}||}z dVW|dS|w)Nr) _WorkerServer)Ztorch._C._distributed_c10dr shutdown)rr serverr V/var/www/auris/lib/python3.10/site-packages/torch/distributed/elastic/control_plane.py_worker_servers  rccsVt}tjt}|dur|t|dVWddS1s$wYdS)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)stackrr r rrs ") rcollections.abcr contextlibrrZ0torch.distributed.elastic.multiprocessing.errorsr__all__rstrrrr r r rs