Function gather_row_token 

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

PURE: the owning token index of a flat (token, slot) row — mirrors the kernel’s row / top_k so the per-token row_adapter gather is verifiable.