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

Merged

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

vraspar merged 7 commits into main from vraspar/fix-gqa-seqlens-k-oob

Add seqlens_k bounds validation in GroupQueryAttention to prevent GEM…

a49b8389

vraspar requested a review from

copilot-pull-request-reviewer 36 days ago

copilot-pull-request-reviewer commented on 2026-04-10

Add non-prompt regression test for seqlens_k underflow guard

6da73057

Address Copilot review: INVALID_ARGUMENT errors, total_seq_len check,…

b2f6dfd0

Add seqlens_k bounds validation in GroupQueryAttention to prevent GEM…

5edfbfc1

vraspar requested a review from

hariharans29 31 days ago

vraspar requested a review from

yuslepukhin 29 days ago

yuslepukhin commented on 2026-04-16

Address review: int64 cast, total_seq_len validation, boundary tests

5d6f96d6

Add SafeInt defense-in-depth and INT32_MAX test for GQA seqlens_k

11f5d838

Use SafeInt for memory allocation size calculation in GQAAttentionBase

d3fddd2f

vraspar requested a review from

yuslepukhin 25 days ago

yuslepukhin approved these changes on 2026-04-20

vraspar enabled auto-merge (squash) 25 days ago

vraspar merged 7c56fa83 into main 25 days ago

vraspar deleted the vraspar/fix-gqa-seqlens-k-oob branch 25 days ago

Reviewers

yuslepukhin

hariharans29

Assignees

No one assigned

Labels

None yet

Milestone

No milestone