Function moe_group_smem_bytes 

pub fn moe_group_smem_bytes(experts: usize, threads: usize) -> usize

The shared memory of lk_moe_group at threads: the running offsets and each warp’s counts.