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

专利信息
申请号: 201911274394.9 申请日: 2019-12-12
公开(公告)号: CN112988367A 公开(公告)日: 2021-06-18
发明(设计)人: 不公告发明人 申请(专利权)人: 中科寒武纪科技股份有限公司
主分类号: G06F9/50 分类号: G06F9/50;G06N3/063
代理公司: 北京华进京联知识产权代理有限公司 11606 代理人: 孙岩
地址: 100190 北京市海*** 国省代码: 北京;11
权利要求书: 查看更多 说明书: 查看更多
摘要:
搜索关键词: 资源 分配 方法 装置 计算机 设备 可读 存储 介质
【权利要求书】:

1.一种资源的分配方法,其特征在于,所述方法应用于多核处理器中,所述方法包括:

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

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

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

2.根据权利要求1所述的方法,其特征在于,所述根据所述各神经节点的连接关系和时序,将所述各神经节点划分为多个神经节点组,包括:

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

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

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

3.根据权利要求1所述的方法,其特征在于,所述根据所述各神经节点的连接关系和时序,将所述各神经节点划分为多个神经节点组,包括:

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

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

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

4.根据权利要求1所述的方法,其特征在于,所述将所述多个神经节点组分配至多核处理器包含的各核中,包括:

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

5.根据权利要求1所述的方法,其特征在于,所述计算图还包括所述各神经节点的运算类型,所述将所述多个神经节点组分配至多核处理器包含的各核中,包括:

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

针对每个神经节点组,如果该神经节点组中包含目标运算类型的神经节点,则在多核处理器包含的核集群中,将未分配有与该神经节点组的时序相同,且包含目标运算类型的神经节点的神经节点组的核集群,确定该神经节点组对应的目标核集群,并将该神经节点组分配至所述目标核集群中当前任务量最小的核中;

如果该神经节点组未包含目标运算类型的神经节点,则将该神经节点组分配至所述多核处理器中当前任务量最小的核中。

6.根据权利要求1所述的方法,其特征在于,所述方法还包括:

根据所述各神经节点的连接关系和时序,确定所述神经节点组对应的前序神经节点组;

针对每个神经节点组,如果该神经节点组与该神经节点组对应的前序神经节点组未分配至同一核中,则在该神经节点组中添加第一同步算子,并在该神经节点组对应的前序神经节点组中添加与所述第一同步算子对应的第二同步算子,所述第一同步算子和与所述第二同步算子用于执行同步运算。

7.一种资源的分配装置,其特征在于,所述装置应用于多核处理器中,所述装置包括:

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

划分模块,用于根据所述各神经节点的连接关系和时序,将所述各神经节点划分为多个神经节点组;

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

下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。

该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于中科寒武纪科技股份有限公司,未经中科寒武纪科技股份有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服

本文链接:http://www.vipzhuanli.com/pat/books/201911274394.9/1.html,转载请声明来源钻瓜专利网。

×

专利文献下载

说明:

1、专利原文基于中国国家知识产权局专利说明书;

2、支持发明专利 、实用新型专利、外观设计专利(升级中);

3、专利数据每周两次同步更新,支持Adobe PDF格式;

4、内容包括专利技术的结构示意图流程工艺图技术构造图

5、已全新升级为极速版,下载速度显著提升!欢迎使用!

请您登陆后,进行下载,点击【登陆】 【注册】

关于我们 寻求报道 投稿须知 广告合作 版权声明 网站地图 友情链接 企业标识 联系我们

钻瓜专利网在线咨询

周一至周五 9:00-18:00

咨询在线客服咨询在线客服
tel code back_top