Add seqlens_k bounds validation in GroupQueryAttention to prevent GEMM OOB#28031

Merged

vraspar merged 7 commits intomainfrom

vraspar/fix-gqa-seqlens-k-oob

Apr 21, 2026

Commits on Apr 10, 2026

Add seqlens_k bounds validation in GroupQueryAttention to prevent GEMM OOB
vraspar
committed
Add non-prompt regression test for seqlens_k underflow guard
vraspar
committed

Commits on Apr 13, 2026

Address Copilot review: INVALID_ARGUMENT errors, total_seq_len check, int64 overflow safety, non-first-prompt test

vraspar
and
Copilot
committed

Commits on Apr 14, 2026

Add seqlens_k bounds validation in GroupQueryAttention to prevent GEMM OOB
vraspar
committed

Commits on Apr 16, 2026

Commits on Apr 20, 2026

Use SafeInt for memory allocation size calculation in GQAAttentionBase
vraspar
committed