基于贪心算法的图神经网络负载均衡方法及装置

本申请提供一种基于贪心算法的图神经网络负载均衡方法及装置，所述方法根据硬件计算单元数量、节点数量及计算负载总量生成滑窗宽度和窗阈值；其中，所述滑窗宽度为每个硬件计算单元分配到的需要计算的矩阵行/列的个数；所述窗阈值为滑窗期望达到的均衡化效果的阈值；将图神经网络中的节点按照计算负载的大小进行排序；基于所述滑窗宽度沿排序后的节点序列执行滑动操作；当窗高度到达所述窗阈值时，停止当前滑窗的滑动；将当前滑窗内节点分配至所述硬件计算单元，以执行聚合计算，解决图神经网络中计算负载不均衡的问题。