Skip to main content

Module spmm

Module spmm 

Source
Expand description

Sparse matrix-dense matrix multiplication (SpMM).

Computes C = alpha * A * B + beta * C where A is a sparse CSR matrix and B, C are dense matrices.

§Strategy

Each warp processes one row of A and tiles across columns of B. Within each row, the warp collaboratively loads non-zero entries and accumulates partial sums into the output columns.

Functions§

spmm
Sparse matrix-dense matrix multiplication: C = alpha * A * B + beta * C.