(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202210868286.X (22)申请日 2022.07.22 (71)申请人 浪潮电子信息产业股份有限公司 地址 250101 山东省济南市高新区浪潮路 1036号 (72)发明人 郝锐　肖麟阁　阚宏伟　 (74)专利代理 机构 北京集佳知识产权代理有限 公司 11227 专利代理师 薛娇 (51)Int.Cl. G06F 9/48(2006.01) G06F 9/50(2006.01) (54)发明名称 任务调度方法、 装置、 电子设备及计算机可 读存储介质 (57)摘要 本申请公开了一种任务调度方法、 装置、 电 子设备及计算机可读存储介质， 应用于RGP U设备 端， 方法包括： 接收主机端下发的任务信息； 根据 所述任务信息确定目标任务、 RGPU调用数量、 当 前RGPU编号； 根据所述RGP U调用数量和所述 当前 RGPU编号从所述目标任务中划分出目标子任务； 执行所述目标子任务。 应用本申请所提供的技术 方案， 可以实现单个任务在多个RGP U上并行执行 的调度， 有效地提高了任务处理效率。 本申请还 公开了另一种应用于主机端的任务调度方法、 装 置、 电子设备即计算机可读存储介质， 同样具有 上述有益效果。 权利要求书2页 说明书11页 附图2页 CN 115098239 A 2022.09.23 CN 115098239 A 1.一种任务调度方法， 其特 征在于， 应用于RGPU设备端， 包括： 接收主机端下发的任务信息； 根据所述任务信息确定目标任务、 RGPU调用数量、 当前RGPU编号； 根据所述RGPU调用数量和所述当前RGPU编号从所述目标任务中划分出目标子任务； 执行所述目标子任务。 2.根据权利要求1所述的任务调度 方法， 其特征在于， 所述根据 所述RGPU调用数量和所 述当前RGPU编号从所述目标任务中划分出目标子任务， 包括： 根据所述RGPU调用数量对所述目标任务进行平均划分， 获得平均划分子任务和余数子 任务； 当所述当前RGPU编号不为最大编号时， 将所述当前RGPU编号对应的平均划分子任务作 为所述目标子任务； 当所述当前RGPU编号为所述最大编号时， 将所述当前RGPU编号对应的平均划分子任务 和所述余数子任务组合 为所述目标子任务。 3.根据权利要求1所述的任务调度方法， 其特 征在于， 所述执 行所述目标子任务， 包括： 确定Core数量、 Warp数量、 Thread数量； 根据所述Core数量、 所述Warp数量、 所述Thread数量将所述目标子任务划分至各个 Thread。 4.一种任务调度方法， 其特 征在于， 应用于主机端， 包括： 根据目标任务确定RGPU调用数量， 并设置各被调用RGPU的RGPU编号； 根据所述目标任务、 所述RGPU调用数量、 所述RGPU编号 生成任务信息； 将所述任务信息下发至各所述被调用RGPU， 以使各所述被调用RGPU根据所述RGPU调用 数量和自身RGPU编号从所述目标任务中划分出目标子任务， 并执 行所述目标子任务。 5.根据权利要求4所述的任务调度 方法， 其特征在于， 所述根据目标任务确定RGPU调用 数量， 包括： 利用API函数查询RGPU可用数量； 结合所述RGPU可用数量和所述目标任务确定所述RGPU调用数量。 6.根据权利要求 4所述的任务调度方法， 其特 征在于， 还 包括： 下发结果获取命令 至各所述被调用RGPU； 接收各所述被调用RGPU根据所述结果获取命令反馈的子任务执 行结果； 根据各所述被调用RGPU的RGPU编号对各所述子任务执行结果进行排列组合， 获得所述 目标任务的执 行结果。 7.一种任务调度装置， 其特 征在于， 应用于RGPU设备端， 包括： 接收模块， 用于 接收主机端下发的任务信息； 确定模块， 用于根据所述任务信息确定目标任务、 RGPU调用数量、 当前RGPU编号； 划分模块， 用于根据所述RGPU调用数量和所述当前RGPU编号从所述目标任务 中划分出 目标子任务； 执行模块， 用于执 行所述目标子任务。 8.一种任务调度装置， 其特 征在于， 应用于主机端， 包括： 设置模块， 用于根据目标任务确定RGPU调用数量， 并设置各被调用RGPU的RGPU编号；权　利　要　求　书 1/2 页 2 CN 115098239 A 2生成模块， 用于根据所述目标任务、 所述RGPU调用数量、 所述RGPU编号 生成任务信息； 下发模块， 用于将所述任务信息下发至各所述被调用RGPU， 以使各所述被调用RGPU根 据所述RGPU调用数量和自身RGPU编号 从所述目标任务中划分出目标子任务， 并执行所述目 标子任务。 9.一种电子设备， 其特 征在于， 包括： 存储器， 用于存 储计算机程序； 处理器， 用于执行所述计算机程序时实现如权利要求1至6任一项所述的任务调度方法 的步骤。 10.一种计算机可读存储介质， 其特征在于， 所述计算机可读存储介质上存储有计算机 程序， 所述计算机程序被处理器执行时实现如权利要求 1至6任一项 所述的任务调 度方法的 步骤。权　利　要　求　书 2/2 页 3 CN 115098239 A 3