一种基于哈希的稀疏矩阵向量乘法优化方法及装置

    公开(公告)号:CN119884572A

    公开(公告)日:2025-04-25

    申请号:CN202411928085.X

    申请日:2024-12-25

    Abstract: 本发明提出一种基于哈希的稀疏矩阵向量乘法优化方法,其特征在于,包括:依据神经网络硬件加速器的硬件结构,对待矩阵向量乘的稀疏矩阵进行划分,得到多个稀疏矩阵块,对该多个稀疏矩阵块进行线性哈希映射,得到待划分矩阵;根据该待划分矩阵的规模及该硬件结构,划分该待划分矩阵,得到多个子矩阵块,并划分该子矩阵块内的并行执行部分和竞争执行部分;神经网络硬件加速器通过对该子矩阵块间竞争执行计算任务,块内并行执行计算任务,得到多个子矩阵计算结果,并通过查找哈希表还原写入的原本顺序;根据该原本顺序合并该多个子矩阵计算结果,得到矩阵向量乘的最终结果。

Patent Agency Ranking