任务调度方法、装置、电子设备及计算机可读存储介质.pdf

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