Skip to main content

spgemm_estimate_memory

singe_cusparse::spgemm

Function spgemm_estimate_memory 

Source 
pub fn spgemm_estimate_memory<Compute: DataTypeLike>(
    ctx: &Context,
    op_a: Operation,
    op_b: Operation,
    alpha: Scalar<'_, Compute>,
    matrix_a: &SparseMatrixDescriptor<'_>,
    matrix_b: &SparseMatrixDescriptor<'_>,
    beta: Scalar<'_, Compute>,
    matrix_c: &mut SparseMatrixDescriptor<'_>,
    algorithm: SpGemmAlgorithm,
    descriptor: &SpGemmDescriptor,
    chunk_fraction: f32,
    external_buffer: Option<DevicePtr>,
) -> Result<(usize, usize)>