Skip to main content

k2_ssd_bmm

Function k2_ssd_bmm 

Source
pub(crate) fn k2_ssd_bmm<B: Backend>(
    c_bnlhr: F<B, 5>,
    b_bnlhr: F<B, 5>,
) -> F<B, 5>
Expand description

Primitive port of crate::mamba2::ssd::serial::k2_ssd_bmm.

Returns the intra-chunk C·Bᵀ block matrix cb_bnhll.