pub const HEAD_TILE_M: usize = 128;
The rows of a tiled per-head matmul block (lk_head_tiled), each block 64 of a head’s outputs.
lk_head_tiled