[发明专利]任务分配方法

说明书

本发明公开了一种任务分配方法。所述任务分配方法包括：向所述数据处理装置的节点芯片发送计算任务命令；数据处理装置的各节点芯片根据各自任务进行运算；数据处理装置向外部控制装置返回运算结果。本发明实施例能够有效避免串联节点芯片之间的数据发送的冲突，通过较少的指令交互实现串联节点芯片的计算任务的分配。

技术领域

本发明涉及数据处理技术领域，特别是涉及一种任务分配方法。

背景技术

目前，随着机器学习尤其是深度学习技术在各领域的应用和发展，对计算装置的数据处理能力提出了更高的要求。GPU处理芯片因为其优于传统CPU的强大的图形处理和并行运算能力，已广泛应用于各领域的数据运算任务，成为通用的深度学习计算平台。

然而，单一GPU架构的计算能力仍然受限，无法满足深度学习、哈希运算等对高强度的数据计算能力的需求。为此，申请号为CN 201610312586.4的中国发明专利申请提出了一种扩展数据处理装置的运算能力的方案，如图1所示。该方案提出了一种由多个节点芯片串联连接的数据处理装置，该数据处理装置经由位于下行通信方向的首节点芯片的外部接口接收数据处理任务，通过串行连接的各级节点芯片对数据处理任务进行运算处理，并通过首节点芯片的外部接口返回数据处理结果。该方案中节点芯片的数量可以根据数据处理任务的运算能力需求进行扩展，并且只需要一个节点芯片与外部设备进行通信连接，不占用外部设备的通信接口，因此，可以提供较强的易于扩展的数据处理能力。

虽然上述现有技术将节点芯片串联连接，每个节点芯片都负责一部分计算处理，加快了数据处理速度，但是各节点芯片之间进行数据发送很容易产生冲突。并且，该数据处理装置接收外部设备传输的数据处理任务，需要在各节点芯片之间分配数据处理任务，如何在多个节点芯片之间分配任务以减少信令的交互也是需要考虑的问题。此外，在串联的各节点芯片处理同一数据处理任务时，可能存在某个节点芯片发生宕机的故障情况，从而造成整个节点芯片组无法正常工作，如何快速进行节点芯片的故障诊断也是需要解决的问题。

发明内容

为了解决上述问题，本发明提出一种任务分配方法。

所述任务分配方法应用于具有多个依次串联连接的节点芯片的数据处理装置，所述方法包括如下步骤：

向所述数据处理装置的节点芯片发送计算任务命令；

数据处理装置的各节点芯片根据各自任务进行运算；

数据处理装置向外部控制装置返回运算结果。

可选地，所述向所述数据处理装置的节点芯片发送计算任务命令，包括：

向所述数据处理装置的节点芯片发送命令，使得节点芯片进入去活状态；

向所述数据处理装置的节点芯片发送地址分配命令，依次为各节点芯片分配芯片地址；

根据各节点芯片的芯片地址，为各节点芯片分配计算任务。

可选地，所述向数据处理装置的节点芯片发送命令，使得节点芯片进入去活状态，包括：

向所述数据处理装置的各节点芯片分别发送命令，使得各节点芯片依次进入去活状态；或者，

向所述数据处理装置的所有节点芯片发送一次命令，使得各节点芯片同时进入去活状态。

可选地，所述依次为各节点芯片分配芯片地址，包括：

当节点芯片处于去活状态时，解析所述地址分配命令，将为其分配的芯片地址保存至寄存器，并转入激活状态；和/或，

当节点芯片处于激活状态时，将所述地址分配命令转发给下一级节点芯片。

可选地，还包括：

节点芯片根据为其分配的计算任务执行哈希运算。