Function dense_gemm_ba_gates_prefill_hopper 

pub fn dense_gemm_ba_gates_prefill_hopper(
    gpu: &dyn GpuBackend,
    kernel: KernelHandle,
    input: DevicePtr,
    ba_weight: &DenseWeight,
    a_log: DevicePtr,
    dt_bias: DevicePtr,
    gate_out: DevicePtr,
    m: u32,
    n: u32,
    k: u32,
    k_stride: u32,
    gate_stride: u32,
    nv: u32,
    vheads_per_group: u32,
    stream: u64,
) -> Result<()>

2026-09-25: Launch dense_gemm_ba_gates_prefill_hopper, one block per token. It takes the parent’s arguments in the parent’s order; only the grid differs.