Function Description

Operator Function

Computes the product of a sparse matrix and a dense matrix. The operator is designed based on the compressed sparse row (CSR) storage structure. It skips zero blocks during loading and computing to maximize the efficiency of computational and memory bandwidth utilization. The core computing kernel is optimized for the Kunpeng platform based on SIMD.

Formula

The SparseGemm operator computes the following matrix multiplication:

, where

• A: sparse matrix
• B: dense matrix
• C: output matrix
• α and β: optional scaling coefficients