Function gather_row_token 

pub fn gather_row_token(row: u32, top_k: u32) -> u32

2026-09-25: The token that owns flat (token, slot) row row: the gather kernels’ row / top_k.