Skip to main content

dot_partial

Function dot_partial 

Source
pub fn dot_partial(
    q_buffer: &str,
    k_buffer: &str,
    accum_var: &str,
    q_base: Expr,
    k_base: Expr,
    d: u32,
) -> Node
Expand description

Emit the dk loop that accumulates a partial dot product into accum_var.