mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
3d541835d5
This adds debug handlers as described in: * https://gist.github.com/d4l3k/828b7be585c7615e85b2c448b308d925 (public copy) * https://docs.google.com/document/d/1la68szcS6wUYElUUX-P6zXgkPA8lnfzpagMTPys3aQ8/edit (internal copy) This is only adding the C++ pieces that will be used from the main process. The Python and torchrun pieces will be added in a follow up PR. This adds 2 handlers out of the box: * `/handler/ping` for testing purposes * `/handler/dump_nccl_trace_pickle` as a POC integration with Flight Recorder Test plan: ``` python test/distributed/elastic/test_control_plane.py ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/126601 Approved by: https://github.com/kurman, https://github.com/c-p-i-o
11 lines
319 B
ReStructuredText
11 lines
319 B
ReStructuredText
Control Plane
|
|
=============
|
|
|
|
.. automodule:: torch.distributed.elastic.control_plane
|
|
.. currentmodule:: torch.distributed.elastic.control_plane
|
|
|
|
This module contains optional helpers that add extra debug and control handlers
|
|
into your application.
|
|
|
|
.. autofunction:: torch.distributed.elastic.control_plane.worker_main
|