[发明专利]资源的分配方法、装置、计算机设备及可读存储介质

说明书

本申请涉及一种资源的分配方法、装置、计算机设备及可读存储介质。该方法应用于多核处理器中，该方法包括：获取目标神经网络模型对应的计算图，所述计算图包括各神经节点的连接关系和时序；根据所述各神经节点的连接关系和时序，将所述各神经节点划分为多个神经节点组；将所述多个神经节点组分配至多核处理器包含的各核中，以使所述多核处理器运行所述目标神经网络模型。采用本申请可以降低多核处理器的通信开销。

技术领域

本申请涉及计算机技术领域，特别是涉及一种资源的分配方法、装置、计算机设备及可读存储介质。

背景技术

目前，多核处理器以其高性能、低功耗的优势，逐步取代传统的单核处理器成为市场的主流。传统的多核处理器在运行神经网络模型时，会将该神经网络模型中的各神经节点随机分配至该多核处理器的各个核中。

然而，随机分配方式将导致不同核之间的通信开销较大。因此，亟需一种能够降低多核处理器通信开销的方案。

发明内容

基于此，有必要针对上述技术问题，提供一种资源的分配方法、装置、计算机设备及可读存储介质。

第一方面，提供了一种资源的分配方法，所述方法应用于多核处理器中，所述方法包括：

获取目标神经网络模型对应的计算图，所述计算图包括各神经节点的连接关系和时序；

根据所述各神经节点的连接关系和时序，将所述各神经节点划分为多个神经节点组；

将所述多个神经节点组分配至多核处理器包含的各核中，以使所述多核处理器运行所述目标神经网络模型。

作为一种可选的实施方式，所述根据所述各神经节点的连接关系和时序，将所述各神经节点划分为多个神经节点组，包括：

根据所述各神经节点的连接关系和时序，确定所述各神经节点对应的前驱神经节点；

针对所述各神经节点中的每个神经节点，如果该神经节点对应的前驱神经节点满足第一预设条件，则创建神经节点组，并将该神经节点添加至创建的神经节点组中，所述第一预设条件为该神经节点不存在对应的前驱神经节点，或者该神经节点对应多个前驱神经节点，或者该神经节点与其他神经节点对应同一前驱神经节点；

如果该神经节点对应的前驱神经节点不满足第一预设条件，则将该神经节点添加至该神经节点对应的前驱神经节点所属的神经节点组中。

作为一种可选的实施方式，所述根据所述各神经节点的连接关系和时序，将所述各神经节点划分为多个神经节点组，包括：

根据所述各神经节点的连接关系和时序，确定所述各神经节点对应的后继神经节点；

针对所述各神经节点中的每个神经节点，如果该神经节点对应的后继神经节点满足第二预设条件，则创建神经节点组，并将该神经节点添加至创建的神经节点组中，所述第二预设条件为该神经节点不存在对应的后继神经节点，或者该神经节点对应多个后继神经节点，或者该神经节点与其他神经节点对应同一后继神经节点；

如果该神经节点对应的后继神经节点不满足第二预设条件，则将该神经节点添加至该神经节点对应的后继神经节点所属的神经节点组中。

作为一种可选的实施方式，所述将所述多个神经节点组分配至多核处理器包含的各核中，包括：

针对每个神经节点组，将该神经节点组分配至多核处理器中当前任务量最小的核中。

作为一种可选的实施方式，所述计算图还包括所述各神经节点的运算类型，所述将所述多个神经节点组分配至多核处理器包含的各核中，包括：

根据所述各神经节点的连接关系和时序，确定所述神经节点组的时序；