pub const ATTN_FOLD_PAIRS: usize = 64;
The value pairs a block of lk_attention_fold folds, a thread each.
lk_attention_fold