pub fn moe_expert_gate_up_act_bf16_grouped(
gpu: &dyn GpuBackend,
kernel: KernelHandle,
input: DevicePtr,
gate_ptrs: DevicePtr,
up_ptrs: DevicePtr,
act: DevicePtr,
expert_offsets: DevicePtr,
sorted_token_ids: DevicePtr,
active_experts: DevicePtr,
active_count: DevicePtr,
sh_gate: DevicePtr,
sh_up: DevicePtr,
sh_act: DevicePtr,
n: u32,
k: u32,
cap: u32,
num_tokens: u32,
stream: u64,
) -> Result<()>Expand description
2026-10-02: Grouped BF16 gate+up and SiLU. gate_ptrs / up_ptrs are device tables of each
routed expert’s [n, k] BF16 weight; sh_gate / sh_up the shared expert’s. Writes the SiLU
product as BF16 hi + lo rows ([positions, 2n] into act, [num_tokens, 2n] into sh_act).