Skip to main content

spgemm_compute

Function spgemm_compute 

Source
pub unsafe fn spgemm_compute<T: SparseScalar>(
    handle: &Handle,
    op_a: Op,
    op_b: Op,
    alpha: &T,
    a: &SpMat<'_, T>,
    b: &SpMat<'_, T>,
    beta: &T,
    c: &mut SpMat<'_, T>,
    alg: SpGEMMAlg,
    plan: &SpGEMMPlan,
    size2: &mut usize,
    buffer2: *mut c_void,
) -> Result<()>
Expand description

Phase 2: compute. Same two-step pattern for buffer2.

ยงSafety

buffer2 must be either null (size-query mode) or a valid device pointer to at least *size2 bytes of writable scratch memory that remains live for the duration of the underlying cuSPARSE call.