(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202210426012.5 (22)申请日 2022.04.22 (71)申请人 湖南大学 地址 410013 湖南省长 沙市岳麓区麓山 南 路 (72)发明人 李肯立　翁万东　肖正　唐卓　 肖国庆　段明星　周旭　廖清　 (74)专利代理 机构 华进联合专利商标代理有限 公司 44224 专利代理师 张思佳 (51)Int.Cl. G06F 9/50(2006.01) G06F 16/901(2019.01) (54)发明名称 计算任务的分配方法、 装置、 计算机设备和 存储介质 (57)摘要 本申请涉及一种计算任务的分配方法、 装 置、 计算机设备和存储介质。 方法包括： 获取计算 任务对应的第一数量个计算图以及各计算子任 务之间的流转关系； 计算每个边界计算子任务的 关键关系增益， 关键关系增益包括第一增益和第 二增益； 根据各边界计算子任务的第一增益和第 二增益调整边界计算子任务至其他计算图， 获得 新的第一数量个计算图； 若新的第一数量个计算 图存在新的边界计算子任务， 返回确定每个边界 计算子任务的关键关系增益的步骤， 直至获得的 新的第一数量个计算图不存在新的边界计算子 任务， 并将每次获得第一数量个计算图之间的关 键关系数量最小的第一数量个计算图， 分配至对 应的计算设备。 采用本方法能够满足实时应用要 求， 减少通信延时。 权利要求书2页 说明书12页 附图6页 CN 114706687 A 2022.07.05 CN 114706687 A 1.一种计算任务的分配方法， 其特 征在于， 所述方法包括： 获取所述计算任务对应的第一数量个计算图以及所述计算任务中各计算子任务之间 的流转关系， 每 个所述计算图包括至少一计算子任务； 计算每个边界计算子任务的关键关系增益， 所述关键关系增益包括第 一增益和第 二增 益， 其中， 所述第一增益根据所述边界计算子任务所属计算图中与所述边界计算子任务存 在的关键关系获得， 所述第二增益根据所述边界计算子任务分别与其他计算图之 间的关键 关系获得； 根据各边界计算子任务的所述第一增益和第二增益调整所述边界计算子任务至其他 计算图， 获得新的第一数量个 计算图； 若新的第一数量个计算图存在新的边界计算子任务， 返回确定每个边界计算子任务的 关键关系增益的步骤， 直至获得 的新的第一数量个计算图不存在新的边界计算子任务， 并 将每次获得第一数量个计算图之间的关键 关系数量最小的第一数量个计算图， 分配至对应 的第一数量个 计算设备。 2.根据权利要求1所述的方法， 其特征在于， 所述计算每个边界计算子任务的关键关系 增益， 所述关键关系增益包括第一增益和第二增益， 包括： 统计所述边界计算子任务所属计算图中与所述边界计算子任务存在关键关系的计算 子任务的数量， 获得所述第一增益； 并统计所述边界计算子任务分别与其他计算图之间存 在关键关系的计算子任务的数量， 获得所述边界计算子任务分别相对其他计算图的第二增 益。 3.根据权利要求1所述的方法， 其特征在于， 所述计算每个边界计算子任务的关键关系 增益， 所述关键关系增益包括第一增益和第二增益， 包括： 将所述边界计算子任务的流 转关系中关键关系赋值 为第一值； 根据所述边界计算子任务所属计算图中与 所述边界计算子任务存在的关键关系， 以及 所述第一值进行计算， 获得所述第一增 益； 并根据所述边界计算子任务分别与其他计算图 之间的关键关系， 以及所述第一 值进行计算， 获得 所述第二增益。 4.根据权利要求1至3任一项所述的方法， 其特征在于， 所述根据各边界计算子任务的 所述第一增益和 第二增益调整 所述边界计算子任务至其他计算图， 获得新的第一数量个计 算图， 包括： 计算每个所述边界计算子任务的所述第二增益和所述第一增益的差值； 将所述差值大于预设阈值的边界计算子任务分配至所述第 二增益对应的计算图， 获得 重新划分后的第一数量个 计算图。 5.根据权利要求4所述的方法， 其特征在于， 所述将所述差值大于预设阈值的边界计算 子任务分配至所述第二增益对应的计算图， 获得重新划分后的第一预设数量个计算图， 包 括： 根据所述流转关系确定所述差值大于预设阈值的边界计算子任务中的目标边界计算 子任务； 将所述目标边界计算子任务分配至所述第 二增益对应的计算图， 获得重新划分后的第 一预设数量个 计算图。 6.根据权利要求1所述的方法， 其特征在于， 所述获取所述计算任务对应的第 一预设数权　利　要　求　书 1/2 页 2 CN 114706687 A 2量个计算图以及所述计算任务中各计算子任务之间的流 转关系， 包括： 获取所述计算任务对应的有向图， 以及执行所述计算任务的计算设备的数量， 所述有 向图包括第二数量个计算子任务， 以及各计算子任务之间的流转关系， 所述流转关系包括 关键关系和非关键关系， 所述第二数量大于所述第一数量； 按照预设规则将所述有向图划分为第一数量个 计算图。 7.根据权利要求6所述的方法， 其特征在于， 所述按照预设规则将所述有向图划分为第 一数量个 计算图， 包括： 根据所述有向图中各计算子任务的流 转关系进行聚类处 理， 获得多个集 合； 将所述多个集 合划分为第一数量个 计算图。 8.根据权利要求7所述的方法， 其特征在于， 所述将所述多个集合划分为第 一数量个计 算图， 包括： 将所述多个集 合随机划分为第一数量个分区； 随机从任一分区中选择任一集合划分到其他分区， 并计算每次随机从任一分区中选择 任一集合划分到其 他分区后， 获得 所有新的分区之间的切边数； 将切边数最小对应的各分区作为所述计算图。 9.一种计算任务的分配装置， 其特 征在于， 所述装置包括： 第一获取模块， 用于获取所述计算任务对应的第 一数量个计算图以及所述计算任务中 各计算子任务之间的流 转关系， 每 个所述计算图包括至少一计算子任务； 第一计算模块， 用于确定每个边界计算子任务的关键关系增益， 所述关键关系增益包 括第一增 益和第二增 益， 其中， 所述第一增益根据所述边界计算子任务所属计算图中与所 述边界计算子任务存在的关键 关系获得， 所述第二增益根据所述边界计算子任务分别与其 他计算图之间的关键关系获得； 调整模块， 用于根据 各边界计算子任务的所述第 一增益和第 二增益调整所述边界计算 子任务至其 他计算图， 获得新的第一数量个 计算图； 分配模块， 用于若新的第一数量个计算图存在新的边界计算子任务， 返回确定每个边 界计算子任务的关键关系增益的步骤， 直至获得的新的第一数量个计算图不存在新的边界 计算子任务， 并将每次获得第一数量个计算图之间的关键 关系数量最小的第一数量个计算 图， 分配至对应的第一数量个 计算设备。 10.一种计算机设备， 包括存储器和 处理器， 所述存储器存储有计算机程序， 其特征在 于， 所述处 理器执行所述计算机程序时实现权利要求1至8中任一项所述的方法的步骤。 11.一种计算机可读存储介质， 其上存储有计算机程序， 其特征在于， 所述计算机程序 被处理器执行时实现权利要求1至8中任一项所述的方法的步骤。权　利　要　求　书 2/2 页 3 CN 114706687 A 3