Skip to main content

spmm_buffer_size

Function spmm_buffer_size 

Source
pub fn spmm_buffer_size<Compute: DataTypeLike>(
    ctx: &Context,
    op_a: Operation,
    op_b: Operation,
    alpha: Scalar<'_, Compute>,
    matrix_a: &SparseMatrixDescriptor<'_>,
    matrix_b: &DenseMatrixDescriptor<'_>,
    beta: Scalar<'_, Compute>,
    matrix_c: &mut DenseMatrixDescriptor<'_>,
    algorithm: SpMmAlgorithm,
) -> Result<usize>