Rate this Page
★★★★★

Control Plane#

Created On: May 30, 2024 | Last Updated On: Jun 04, 2024

This module contains optional helpers that add extra debug and control handlersinto your application.

torch.distributed.elastic.control_plane.worker_main()[source]#

This is a context manager that wraps your main entry function. This combinesthe existingerrors.record logic as well as a new_WorkerServer thatexposes handlers via a unix socket specified byTorch_WORKER_SERVER_SOCKET.

Example

@worker_main()defmain():passif__name__=="__main__":main()
Return type:

Generator[None, None, None]