[发明专利]任务调度方法、装置、电子设备及计算机可读存储介质

权利要求书

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编号；

生成模块，用于根据所述目标任务、所述RGPU调用数量、所述RGPU编号生成任务信息；

下发模块，用于将所述任务信息下发至各所述被调用RGPU，以使各所述被调用RGPU根据所述RGPU调用数量和自身RGPU编号从所述目标任务中划分出目标子任务，并执行所述目标子任务。

9.一种电子设备，其特征在于，包括：

存储器，用于存储计算机程序；

处理器，用于执行所述计算机程序时实现如权利要求1至6任一项所述的任务调度方法的步骤。

10.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质上存储有计算机程序，所述计算机程序被处理器执行时实现如权利要求1至6任一项所述的任务调度方法的步骤。