鲲鹏社区首页
中文
注册
开发者
我要评分
获取效率
正确性
完整性
易理解
在线提单
论坛求助

功能描述

算子功能

用于计算稀疏矩阵与稠密矩阵的乘积。算子基于压缩稀疏行(CSR)存储结构设计,通过在装载与计算阶段跳过零块,实现计算与访存的高效利用。核心计算内核针对鲲鹏平台进行了SIMD优化。

算子公式

SparseGemm算子用于计算如下形式的矩阵乘法:

其中:

  • A:稀疏矩阵
  • B:稠密矩阵
  • C:输出矩阵
  • α,β:可选缩放系数