pub fn mamba2_ssd_scan(
gpu: &dyn GpuBackend,
kernel: KernelHandle,
h_state: DevicePtr,
x: DevicePtr,
b_proj: DevicePtr,
c_proj: DevicePtr,
d_param: DevicePtr,
dt_f32: DevicePtr,
da_cs: DevicePtr,
cb: DevicePtr,
output: DevicePtr,
seq_len: u32,
num_heads: u32,
head_dim: u32,
state_size: u32,
n_groups: u32,
nchunks: u32,
batch_size: u32,
x_stride: u32,
bc_stride: u32,
y_stride: u32,
stream: u64,
) -> Result<()>Expand description
2026-09-25: K3: chunk_state, state_passing and chunk_scan fused in one kernel. The running state h0 stays in shared memory, so no per-chunk state tensor is written to global memory.