上海思朗科技股份有限公司

本申请提供一种椭球3D‑FFT的任务分配方法、装置、设备、介质，该方法根据频域数据确定计算顺序；根据计算顺序构建pencil任务；通过二分答案算法确定分段代价上限，并以贪心策略对pencil任务构成的任务序列进行分段，将pencil任务分段下发给对应计算组；沿计算顺序的首个方向均匀划分slab任务，将slab任务分段下发给对应计算组。本申请在数据稀疏、任务不均匀的阶段，通过二分答案算法和贪心策略进行pencil任务划分，实现多计算组之间的动态负载均衡，在数据逐渐变为规则稠密的阶段，采用固定长度均匀划分slab任务，使得通信与计算能够形成稳定流水线，提升整体带宽利用率，提高通信与计算重叠效率。

本申请提供一种SpGEMM动态调度方法、装置、设备、存储介质，该方法包括：提取待相乘的稀疏矩阵的关键特征和硬件平台信息；根据关键特征过滤累加器；根据硬件平台信息确定计算范式；基于过滤后的累加器和计算范式，进行稀疏通用矩阵乘法SpGEMM调度。本申请的方法，根据待相乘的稀疏矩阵的关键特征和硬件平台信息进行自动的累加器的过滤和计算范式的确定，提升了SpGEMM调度方案通用性和SpGEMM调度性能的稳定性，确保SpGEMM调度的高精度和强鲁棒性。

本申请实施例中提供了一种图形处理器集群的通信方法、装置、计算机设备和存储介质，涉及计算机应用技术领域。该方法包括：针对第一张量并行分组中每个第一目标图形处理器，预先为第一目标图形处理器与第二张量并行分组中每个第二目标图形处理器构建第二通信链路；其中，第一张量并行分组是多个张量并行分组中的任一张量并行分组，第二张量并行分组是多个张量并行分组中与第一张量并行分组不同的张量并行分组；控制第一张量并行分组中的第一子图形处理器将内部存储的待归约数据划分为多个待归约子数据，并将多个待归约子数据分别基于第一通信链路和第二通信链路传输至第一张量并行分组中的第二子图形处理器。该方式可提高带宽使用率及通信效率。