Skip to content

Add quantized KV attention primitives - #25033

Open
xingguo01 wants to merge 1 commit into
pytorch:mainfrom
xingguo01:arm-quantized-kv-attention-primitives
Open

xingguo01 wants to merge 1 commit into
pytorch:mainfrom
xingguo01:arm-quantized-kv-attention-primitives

Conversation

@xingguo01

@xingguo01 xingguo01 commented Nov 13, 2026 •

Copy link
Copy Markdown
Collaborator

Configure native SDPA through quantizer annotations and decomposition provenance instead of requiring model-specific attention arithmetic. Support A16 attention and experimental A8 QK/AV, deriving INT32 QK scales from symmetric operands and retaining A16 masked softmax.

Keep calibrated integer cache storage separate from attention policy. Retain the experimental Vela-specific INT16 variable path without a model-specific SDPA source transform. Cover quantizer selection, masks, GQA, accumulator bounds, and delegated mutable-buffer ownership.

AI-assisted-by: Codex

cc @digantdesai @freddan80 @per @zingo @oscarandersson8218 @mansnils @Sebastian-Larsson @robell @rascani

Configure native SDPA through quantizer annotations and decomposition
provenance instead of requiring model-specific attention arithmetic.
Support A16 attention and experimental A8 QK/AV, deriving INT32 QK
scales from symmetric operands and retaining A16 masked softmax.

Keep calibrated integer cache storage separate from attention policy.
Retain the experimental Vela-specific INT16 variable path without a
model-specific SDPA source transform. Cover quantizer selection, masks,
GQA, accumulator bounds, and delegated mutable-buffer ownership.

AI-assisted-by: Codex
Change-Id: Ia0d7d8113e3d0f6aa6b625a121329c86d7b6c539
Signed-off-by: Xingguo Li <xingguo.li@arm.com>
@pytorch-bot

pytorch-bot Bot commented Nov 13, 2026 •

Copy link
Copy Markdown

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/25033

Note: Links to docs will display an error until the docs builds have been completed.

⏳ No Failures, 217 Pending

As of commit ab23fc2 with merge base 765e973 (image):
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@meta-cla meta-cla Bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Nov 13, 2026
@github-actions github-actions Bot added ciflow/trunk module: arm Issues related to arm backend labels Nov 13, 2026
@xingguo01 xingguo01 added release notes: arm Changes to the ARM backend delegate help wanted Extra attention is needed labels Nov 13, 2026

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/trunk CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. help wanted Extra attention is needed module: arm Issues related to arm backend release notes: arm Changes to the ARM backend delegate

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant