flashinfer.msa_ops.msa_icp_index_score_columns

flashinfer.msa_ops.msa_icp_index_score_columns(max_seq_len: int, page_size: int) → int

Prefill score columns: ceil(max_seq_len / 128) for P128, half of it (rounded up) for compact P256 scores.