w4a16_gemm_n128_m128_v3

spark_model::layers::ops

Function w4a16_gemm_n128_m128_v3 

Source 
pub fn w4a16_gemm_n128_m128_v3(
    gpu: &dyn GpuBackend,
    kernel: KernelHandle,
    input: DevicePtr,
    weight: &QuantizedWeight,
    output: DevicePtr,
    m: u32,
    n: u32,
    k: u32,
    stream: u64,
) -> Result<()>
Expand description

W4A16 GEMM v3: MiniMax-only shadow with K_STEP=64 (was 32 in v2). Halves K-iteration count; doubles per-iter MMA count. 1 CTA/SM (was 3 for v2) due to larger SMEM footprint.