bf16_gemm_act_weight_t

Function bf16_gemm_act_weight_t 

Source
pub fn bf16_gemm_act_weight_t(
    act: u64,
    weight: u64,
    out: u64,
    m: u32,
    n: u32,
    k: u32,
    stream: u64,
) -> Result<()>
Expand description

Row-major out[M,N] = act[M,K] @ weight[N,K]ᵀ, all BF16 — the standard projection GEMM (activation × transposed weight). Maps to cuBLASLt’s column-major convention as D[N,M] = opT(weightᶜ[K,N]) · opN(actᶜ[K,M]).