一种稀疏矩阵乘法算子的数据处理加速方法和装置

    公开(公告)号:CN119474631A

    公开(公告)日:2025-02-18

    申请号:CN202411532329.2

    申请日:2024-10-30

    Abstract: 本发明提供一种稀疏矩阵乘法算子的数据处理加速方法和装置,包括:数据处理过程中获得需处理的稀疏矩阵乘法算子,对稀疏矩阵进行压缩后其存储形式包括非零元数组、行指针数组、全局列索引数组和本地列索引数组。全局列索引记录稀疏矩阵划分的一组连续预设数量的稀疏行中非零元素在稀疏矩阵的列索引,每个列索引数组中全局列索引不重复。本地列索引记录每个非零元素在全局列索引数组中的位置索引。利用全局列索引访问稠密矩阵中对应的行数据并读取到临时缓存区。利用本地列索引访问临时缓存区内稠密矩阵中对应行数据。计算非零元素与稠密矩阵中对应行数据的乘积并累加,输出结果矩阵。本发明能够提高数据处理速度,降低内存访问开销。

Patent Agency Ranking