Skip to main content

SPARSE_SPGEMM

Constant SPARSE_SPGEMM 

Source
pub const SPARSE_SPGEMM: &str = "rlx_sparse.spgemm";
Expand description

Sparse-sparse matrix multiply (CSR × CSR → CSR). 6 inputs: (a_values, a_col_idx, a_row_ptr, b_values, b_col_idx, b_row_ptr) plus attrs encoding (k: u32 = inner dim = b’s row count). Output is a packed buffer [c_values | c_col_idx_as_f64 | c_row_ptr_as_f64] with sizes encoded in attrs alongside k. v1 caps nnz output at max_nnz (attrs); allocate generously for known patterns.