【技术实现步骤摘要】
资源的分配方法、装置、计算机设备及可读存储介质
本申请涉及计算机
,特别是涉及一种资源的分配方法、装置、计算机设备及可读存储介质。
技术介绍
目前,多核处理器以其高性能、低功耗的优势,逐步取代传统的单核处理器成为市场的主流。传统的多核处理器在运行神经网络模型时,会将该神经网络模型中的各神经节点随机分配至该多核处理器的各个核中。然而,随机分配方式将导致不同核之间的通信开销较大。因此,亟需一种能够降低多核处理器通信开销的方案。
技术实现思路
基于此,有必要针对上述技术问题,提供一种资源的分配方法、装置、计算机设备及可读存储介质。第一方面,提供了一种资源的分配方法,所述方法应用于多核处理器中,所述方法包括:获取目标神经网络模型对应的计算图,所述计算图包括各神经节点的连接关系和时序;根据所述各神经节点的连接关系和时序,将所述各神经节点划分为多个神经节点组;将所述多个神经节点组分配至多核处理器包含的各核中,以使所述多核处理器运行所述目标神经网络模型。作为一种可选 ...
【技术保护点】
1.一种资源的分配方法,其特征在于,所述方法应用于多核处理器中,所述方法包括:/n获取目标神经网络模型对应的计算图,所述计算图包括各神经节点的连接关系和时序;/n根据所述各神经节点的连接关系和时序,将所述各神经节点划分为多个神经节点组;/n将所述多个神经节点组分配至多核处理器包含的各核中,以使所述多核处理器运行所述目标神经网络模型。/n
【技术特征摘要】
1.一种资源的分配方法,其特征在于,所述方法应用于多核处理器中,所述方法包括:
获取目标神经网络模型对应的计算图,所述计算图包括各神经节点的连接关系和时序;
根据所述各神经节点的连接关系和时序,将所述各神经节点划分为多个神经节点组;
将所述多个神经节点组分配至多核处理器包含的各核中,以使所述多核处理器运行所述目标神经网络模型。
2.根据权利要求1所述的方法,其特征在于,所述根据所述各神经节点的连接关系和时序,将所述各神经节点划分为多个神经节点组,包括:
根据所述各神经节点的连接关系和时序,确定所述各神经节点对应的前驱神经节点;
针对所述各神经节点中的每个神经节点,如果该神经节点对应的前驱神经节点满足第一预设条件,则创建神经节点组,并将该神经节点添加至创建的神经节点组中,所述第一预设条件为该神经节点不存在对应的前驱神经节点,或者该神经节点对应多个前驱神经节点,或者该神经节点与其他神经节点对应同一前驱神经节点;
如果该神经节点对应的前驱神经节点不满足第一预设条件,则将该神经节点添加至该神经节点对应的前驱神经节点所属的神经节点组中。
3.根据权利要求1所述的方法,其特征在于,所述根据所述各神经节点的连接关系和时序,将所述各神经节点划分为多个神经节点组,包括:
根据所述各神经节点的连接关系和时序,确定所述各神经节点对应的后继神经节点;
针对所述各神经节点中的每个神经节点,如果该神经节点对应的后继神经节点满足第二预设条件,则创建神经节点组,并将该神经节点添加至创建的神经节点组中,所述第二预设条件为该神经节点不存在对应的后继神经节点,或者该神经节点对应多个后继神经节点,或者该神经节点与其他神经节点对应同一后继神经节点;
如果该神经节点对应的后继神经节点不满足第二预设条件,则将该神经节点添加至该神经节点对应的后继神经节点所属的神经节点组中。
4.根据权利要求1所述的方法,其特征在于,所述将所述多个神经节点组分配至多核处理器包含的各核中,包括:
针对每个神经节点组,将该神经节点组分配至多核处理器中当前任务量最小的核中。
5.根据权利要求1所述的方法,其特征在于,所述计算图还包括所述...
【专利技术属性】
技术研发人员:不公告发明人,
申请(专利权)人:中科寒武纪科技股份有限公司,
类型:发明
国别省市:北京;11
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。