事件¶
模块包含与标准 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.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, ... )