[发明专利]基于图形处理器的图计算节点向量负载平衡方法及装置

权利要求书

1.一种基于图形处理器的图计算节点向量负载平衡方法，其特征在于，所述图形处理器的控制处理器进行负载平衡的调度步骤包括：

1）图形处理器的控制处理器定期找出所有SM核中最长的和最短的节点向量长度并计算差值；

2）图形处理器的控制处理器判断计算得到的差值是否超过预设阈值，如果尚未超过预设阈值，则结束并退出；否则跳转执行下一步；

3）图形处理器的控制处理器选择需要进行负载平衡的两个SM核，将两个SM核的状态分别设置为等待捐赠负载状态和接收负载捐赠的状态，并计算需要捐赠的节点数量；

4）图形处理器的控制处理器向状态为接收负载捐赠的SM核发送负载平衡请求，且在该SM核在全局存储中申请存储空间地址成功后跳转执行下一步；

5）图形处理器的控制处理器通知状态为等待捐赠负载的SM核开始转移节点向量的数据，被转移的节点向量数据会被写入负载接收方申请的存储空间地址中，完成节点向量的负载平衡；

步骤4）图形处理器的控制处理器向状态为接收负载捐赠的SM核发送负载平衡请求以及步骤5）中图形处理器的控制处理器通知状态为等待捐赠负载的SM核开始转移节点向量的数据均为通过SM核的控制寄存器实现的，所述SM核的控制寄存器包括节点向量长度寄存器R_VLS、负载平衡控制寄存器R_BC、负载平衡状态寄存器R_BS以及两个负载平衡工作寄存器R_BW0和R_BW1，节点向量长度寄存器R_VLS用于记录SM核中的节点向量长度，负载平衡控制寄存器R_BC用于记录等待捐赠负载状态、接收负载捐赠的状态、正在捐赠节点的状态，负载平衡状态寄存器R_BS用于记录地址计算状态，负载平衡工作寄存器R_BW0用于记录全局存储中申请的存储空间地址，负载平衡工作寄存器R_BW1用于记录需要捐赠的节点数量。

2.根据权利要求1所述的基于图形处理器的图计算节点向量负载平衡方法，其特征在于，步骤3）中计算需要捐赠的节点数量的值为两个SM核的输出节点向量长度之间的差值的二分之一。

3.一种基于图形处理器的图计算节点向量负载平衡装置，其特征在于，该装置被编程以执行权利要求1或2所述基于图形处理器的图计算节点向量负载平衡方法的步骤。

4.一种图形处理器，包括控制处理器和多个SM核，其特征在于，该控制处理器被编程以执行权利要求1或2所述基于图形处理器的图计算节点向量负载平衡方法的步骤。