pub const ATTN_SHARED_STEP: usize = 16;
The cached rows lk_attention_shared stages a step (LK_ST in its source).
lk_attention_shared
LK_ST