def _get_debug_info():
from . import _rref_context_get_debug_info
from .api import _agent
info = _rref_context_get_debug_info()
info.update(_agent.get_debug_info())
info.update(dist_autograd._get_debug_info())
return info
def test_debug_info(self):
# only test keys in this test case. Values should be covered by
# individual module debug info tests
from torch.distributed.rpc import (_get_debug_info,
_rref_context_get_debug_info)
from torch.distributed.rpc.api import _agent
import torch.distributed.autograd as dist_autograd
info = _get_debug_info()
rref_info = _rref_context_get_debug_info()
agent_info = _agent.get_debug_info()
autograd_info = dist_autograd._get_debug_info()
common_keys = rref_info.keys() & agent_info.keys(
) & autograd_info.keys()
self.assertEqual(0, len(common_keys))
expected = {}
expected.update(rref_info)
expected.update(agent_info)
expected.update(autograd_info)
self.assertEqual(expected, info)
def _get_debug_info():
from . import _rref_context_get_debug_info
info = _rref_context_get_debug_info()
info.update(api._get_current_rpc_agent().get_debug_info())
info.update(dist_autograd._get_debug_info())
return info
