整体架构

断点续训特性基于VolcanoAscend OperatorAscend Device PluginNodeDClusterDElastic Agent等组件,整体的架构如图1所示。

图1 整体架构

其中各组件的能力如下:

  1. Ascend Device Plugin:故障发现组件,提供NPU资源管理、NPU芯片故障和NPU网络故障上报等能力。
  2. NodeD:故障发现组件,提供节点健康状态上报能力;同时,提供节点硬件故障上报能力。
  3. Volcano:故障处理组件,提供故障任务重调度的能力。
  4. Ascend Operator:为分布式训练任务创建pod生成环境变量;提供静态组网集合通信所需的ranktable信息。
  5. ClusterD:获取集群中所有Ascend Device PluginNodeD上报的数据,整理后发送给Volcano
  6. Elastic Agent:提供与K8s集群的训练集群控制中心的通信功能,完成故障修复、恢复训练。
  7. 训练模型代码:需要进行断点续训相关能力的适配操作。