webgpu: Enable FlashAttention for batched GQA with right-padded prompts - #29247

Merged

Jiajia Qin (qjia7) merged 8 commits into

microsoft:mainfrom

qjia7:feat/webgpu-flash-attention-batched

Jun 26, 2026

Commits on Jun 11, 2026

webgpu: fix GQA batched right-padded prefill with do_rotary
Jiajia Qin (qjia7)
committed
591df5b
webgpu: support batch_size > 1 on the FlashAttention path
Jiajia Qin (qjia7)
committed
4c0492c

Commits on Jun 24, 2026

Merge branch 'main' into feat/webgpu-flash-attention-batched
Jiajia Qin (qjia7)
committed
849ef98

Commits on Jun 25, 2026