Skip to main content

gemm_with

Function gemm_with 

Source
pub fn gemm_with<T: GemmScalar>(
    ws: &mut Workspace,
    alpha: T,
    a: MatRef<'_, T>,
    b: MatRef<'_, T>,
    beta: T,
    c: MatMut<'_, T>,
    par: Parallelism,
)
Expand description

Like gemm but reuses a caller-owned Workspace: zero heap allocation once the workspace has grown to fit the 1st sufficiently large call

ยงPanics

Same conditions as gemm