在审

一种用于稀疏矩阵乘的数据处理方法、系统、设备和存储介质

王庆林、洪楚河、刘杰、刘珂、刘根程、彭兴、陈新海、张翔、王柄霖、张庆阳、杨博
中国人民解放军国防科技大学

摘要

本发明提供的一种用于稀疏矩阵乘的数据处理方法,包括以下步骤:基于第一稀疏矩阵的处理单元的非零元素数量、第二稀疏矩阵中对应的处理单元的非零元素数量和第一预设选择策略确定第一目标累加器,基于第一目标累加器进行模拟矩阵乘,得到输出矩阵的对应输出单元的非零元素数量;基于第一稀疏矩阵的处理单元的非零元素数量、输出矩阵的对应输出单元的非零元素数量和第二预设选择策略确定第二目标累加器;目标累加器为至少包括归并累加器的混合累加器;利用第二目标累加器对第一稀疏矩阵的处理单元中的非零元素与第二稀疏矩阵中对应的处理单元中的非零元素的乘积进行累加求和,得到输出矩阵的对应输出单元。本申请提升了稀疏矩阵乘的运算性能。

暂无引用专利