Control Plane

创建于：2024年5月30日 | 最后更新于：2024年6月4日

此模块包含可选的助手，可为您的应用程序添加额外的调试和控制处理程序。

torch.distributed.elastic.control_plane.worker_main()

这是一个包装您主入口函数的上下文管理器。它结合了现有的 errors.record 逻辑以及一个新的 _WorkerServer，该服务器通过由 Torch_WORKER_SERVER_SOCKET 指定的 Unix 套接字公开处理程序。

示例

@worker_main()
def main():
    pass


if __name__ == "__main__":
    main()

返回类型

Generator[None, None, None]