pub fn kquant_mmq_gemm(
gpu: &dyn GpuBackend,
kernel_nc: KernelHandle,
kernel_wc: KernelHandle,
a_q8: DevicePtr,
w_blocks: DevicePtr,
out_bf16: DevicePtr,
m: u32,
n: u32,
k: u32,
smem: u32,
stream: u64,
) -> Result<()>