onnxruntime
[CUDA] Extend DynamicSparseAttention INT8 KV cache and PackedSparseAttentionIndexer QSA
#33472
Merged

[CUDA] Extend DynamicSparseAttention INT8 KV cache and PackedSparseAttentionIndexer QSA #33472

apsonawane
apsonawane Extend sparse attention caches and optimize packed indexing
52c73b51
apsonawane Align QSA position overrides and CUDA-only outputs across providers
7be8d682
apsonawane Update generated QSA position override documentation
b538d1fc
apsonawane apsonawane requested a review from copilot-pull-request-reviewer copilot-pull-request-reviewer 12 days ago
copilot-pull-request-reviewer
copilot-pull-request-reviewer commented on 2026-10-19
apsonawane Fix sparse attention workspace and optional interface handling
538bd982
apsonawane apsonawane marked this pull request as ready for review 12 days ago
apsonawane apsonawane requested a review from copilot-pull-request-reviewer copilot-pull-request-reviewer 12 days ago
copilot-pull-request-reviewer
copilot-pull-request-reviewer commented on 2026-10-19
apsonawane Fix WebGPU binding-limit test and clarify mixed key-state support
93565558
apsonawane
apsonawane apsonawane enabled auto-merge (squash) 12 days ago
apsonawane Add compact QSA selection and specialized INT8 attention dispatch
d21946e6
apsonawane
apsonawane
apsonawane Merge main into asonawane/main-ops-sparse-attention-indexer
b42576e6
apsonawane
github-actions
github-actions commented on 2026-10-22
apsonawane Apply lintrunner formatting to PR changes
b71498b2
apsonawane
tianleiwu Enable default-on sparse attention optimizations, S<=8 grouped decode…
c0c4e7dd
apsonawane apsonawane added model:qwen
tianleiwu [CUDA] Fold #33546 and S=4 prefix-state outputs (mid_kv_buffer2/3, mi…
6548c21c
tianleiwu Merge remote-tracking branch 'origin/main' into tlwu/pr33472-updates
58b6da62
github-actions
github-actions commented on 2026-10-23
tianleiwu Apply lintrunner formatting
3b52694c
tianleiwu [CUDA] Revert unused QSA position_ids and WebGPU shader changes in Pa…
5ed039c7
tianleiwu tianleiwu changed the title Extend sparse attention caches and optimize packed indexing [CUDA] Extend DynamicSparseAttention INT8 KV cache and PackedSparseAttentionIndexer QSA 7 days ago
github-actions
github-actions commented on 2026-10-24
tianleiwu fix(cuda): fix shared-provider include order, Windows macro clash, te…
6511e8a3
tianleiwu Merge remote-tracking branch 'origin/main' into pr33472-work
a9afd694
tianleiwu Optimize GroupedDecodeDynamicSparseAttentionKernel for 12 query heads…
a523af45
tianleiwu Remove legacy mid-state snapshot outputs from PackedSparseAttentionIn…
61c5d87f
tianleiwu tianleiwu requested a review from copilot-pull-request-reviewer copilot-pull-request-reviewer 5 days ago
copilot-pull-request-reviewer
copilot-pull-request-reviewer commented on 2026-10-26
tianleiwu perf(cuda): optimize INT8 KV DynamicSparseAttention prefill and Packe…
aa787761
tianleiwu fix(cuda): fix FP16 ratio-12 grouped decode MMA and expand prefill te…
18a13c70
tianleiwu fix(cuda): avoid narrowing in sparse attention test
fea46be4
tianleiwu [CUDA] Support per-batch committed_length in DynamicSparseAttention (…
9116cb9d
apsonawane apsonawane requested a review from copilot-pull-request-reviewer copilot-pull-request-reviewer 5 days ago
copilot-pull-request-reviewer
copilot-pull-request-reviewer commented on 2026-10-26
apsonawane
apsonawane commented on 2026-10-26
apsonawane
apsonawane commented on 2026-10-26
tianleiwu [CUDA] DSA: fail strict validation on inconsistent committed cursor
a4151cc4
tianleiwu test(cuda): run alternate DSA/indexer switches in fresh processes
977e9813
tianleiwu docs: document DSA/indexer dispatch switches, INT8 encoding and statu…
0639efa7
tianleiwu [CUDA] Remove packed sparse indexer radix top-K path
cd4e2e42
jiafatom
kunal-vaishnavi
kunal-vaishnavi commented on 2026-10-27
tianleiwu fix(cuda): guard DSA wide loads on buffer alignment and fix iOS test …
c8c7b77c
tianleiwu
tianleiwu tianleiwu requested a review from kunal-vaishnavi kunal-vaishnavi 4 days ago
jiafatom
jiafatom approved these changes on 2026-10-28
apsonawane apsonawane merged a23131fe into main 3 days ago
apsonawane apsonawane deleted the asonawane/main-ops-sparse-attention-indexer branch 3 days ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone