快捷方式

事件

模块包含与标准 Python 日志记录集成的事件处理机制。

用法示例

from torch.distributed.elastic import events
event = events.Event(name="test_event", source=events.EventSource.WORKER, metadata={...})
events.get_logging_handler(destination="console").info(event)

API 方法

torch.distributed.elastic.events.record(event, destination='null')[source]
torch.distributed.elastic.events.construct_and_record_rdzv_event(run_id, message, node_state, name='', hostname='', pid=None, master_endpoint='', local_id=None, rank=None)[source]

初始化 rendezvous 事件对象并记录其操作。

参数
  • run_id (str) – rendezvous 的运行 ID。

  • message (str) – 描述事件的消息。

  • node_state (NodeState) – 节点的状态 (INIT、RUNNING、SUCCEEDED、FAILED)。

  • name (str) – 事件名称。(例如,正在执行的当前操作)。

  • hostname (str) – 节点的主机名。

  • pid (Optional[int]) – 节点的进程 ID。

  • master_endpoint (str) – 如果已知,则为 rendezvous 存储的 master 端点。

  • local_id (Optional[int]) – 如果在 dynamic_rendezvous.py 中定义,则为节点的 local_id

  • rank (Optional[int]) – 如果已知,则为节点的秩。

返回值

返回类型

示例

>>> # See DynamicRendezvousHandler class
>>> def _record(
...     self,
...     message: str,
...     node_state: NodeState = NodeState.RUNNING,
...     rank: Optional[int] = None,
... ) -> None:
...     construct_and_record_rdzv_event(
...         name=f"{self.__class__.__name__}.{get_method_name()}",
...         run_id=self._settings.run_id,
...         message=message,
...         node_state=node_state,
...         hostname=self._this_node.addr,
...         pid=self._this_node.pid,
...         local_id=self._this_node.local_id,
...         rank=rank,
...     )
torch.distributed.elastic.events.get_logging_handler(destination='null')[source]
返回类型

处理程序

事件对象

class torch.distributed.elastic.events.api.Event(name, source, timestamp=0, metadata=<factory>)[source]

该类表示在 torchelastic 作业执行期间发生的通用事件。

事件可以是任何有意义的操作。

参数
  • name (str) – 事件名称。

  • (事件源) – 事件生成器,例如代理或工作进程

  • 时间戳 (int) – 事件发生时的毫秒级时间戳。

  • 元数据 (Dict[str, Optional[Union[str, int, float, bool]]]) – 与事件关联的其他数据。

class torch.distributed.elastic.events.api.EventSource(value)[source]

事件生成器的已知标识符。

torch.distributed.elastic.events.api.EventMetadataValue

Optional[Union[str, int, float, bool]] 的别名

文档

获取 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源