Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
26 commits
Select commit Hold shift + click to select a range
d66c7a8
[QNN EP] Add support for Attention Op
qti-mbadnara Jul 17, 2026
a0a2ce3
[QNN EP] Add support for Attention Op
qti-mbadnara Jul 17, 2026
5f0ef18
Update documentation
qti-mbadnara Jul 17, 2026
7ae419f
Merge branch 'main' into dev/qti-mbadnara/add-support-for-attention-op
qti-mbadnara Jul 17, 2026
e8b97ce
Lint fixes
qti-mbadnara Jul 17, 2026
2d6a7d3
Update tolerance to 2e-3f
qti-mbadnara Jul 20, 2026
2093cf0
Decomposition fix
qti-mbadnara Jul 20, 2026
d9e4865
Merge branch 'main' into dev/qti-mbadnara/add-support-for-attention-op
qti-mbadnara Jul 20, 2026
734f701
Build fix
qti-mbadnara Jul 20, 2026
24fc679
Gate Softcap tests to ARM64
qti-mbadnara Jul 20, 2026
974530d
Merge branch 'main' into dev/qti-mbadnara/add-support-for-attention-op
qti-mbadnara Jul 22, 2026
5d91252
[QNN EP] Use native GQA op for GPU
qti-mbadnara Jul 23, 2026
9a04ab4
Act on review comments
qti-mbadnara Jul 30, 2026
b7553f3
CI fixes
qti-mbadnara Jul 30, 2026
3c15d60
[QNN EP] Fix build failures
qti-mbadnara Jul 30, 2026
8411dd9
Act on review comments
qti-mbadnara Jul 30, 2026
68cf288
Merge branch 'main' into dev/qti-mbadnara/add-support-for-attention-op
qti-mbadnara Jul 30, 2026
f522c24
Fixes
qti-mbadnara Jul 30, 2026
8c6fc24
[QNN EP] Fix CI build failures
qti-mbadnara Jul 31, 2026
cb80676
CI fix
qti-mbadnara Jul 31, 2026
4e2b319
[QNN EP] Act on review comments
qti-mbadnara Aug 5, 2026
fc22e1b
[QNN EP] Nit
qti-mbadnara Aug 5, 2026
6e6631f
[QNN EP] Switch back to coarse ops
qti-mbadnara Aug 7, 2026
d1d87d1
[QNN EP] Act on review comments
qti-mbadnara Aug 11, 2026
267936b
Merge branch 'main' into dev/qti-mbadnara/add-support-for-attention-op
qti-mbadnara Aug 11, 2026
de79040
Merge branch 'main' into dev/qti-mbadnara/add-support-for-attention-op
qti-mbadnara Aug 11, 2026
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions docs/execution_providers/QNN-ExecutionProvider.md
Original file line number Diff line number Diff line change
Expand Up @@ -509,6 +509,7 @@ ort.unregister_execution_provider_library(ep_registration_name)
|ai.onnx:ArgMin||
|ai.onnx:Asin||
|ai.onnx:Atan||
|ai.onnx:Attention|Opsets 23–24. 3D \[B,S,n·hs\] and 4D \[B,n,S,hs\] inputs. HTP: full decomposition for MHA, GQA/MQA, is\_causal, attn\_mask, softcap, KV cache, qk\_matmul\_output modes 0–3. GPU: native QNN\_OP\_GROUP\_QUERY\_ATTENTION when is\_causal=1 and no softcap/attn\_mask/qk\_output (covers MHA and GQA/MQA); decomposition otherwise. Static shapes only.|
|ai.onnx:AveragePool||
|ai.onnx:Bernoulli|CPU and HTP backends supported.|
|ai.onnx:BatchNormalization|fp16 supported since 1.18.0|
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@ namespace qnn {
OpBuilderRegistrations::OpBuilderRegistrations() {
CreateArgMaxMinOpBuilder("ArgMax", *this);
CreateArgMaxMinOpBuilder("ArgMin", *this);
CreateAttentionOpBuilder("Attention", *this);
CreateBatchNormalizationOpBuilder("BatchNormalization", *this);
CreateBernoulliOpBuilder("Bernoulli", *this);
CreateCastOpBuilder("Cast", *this);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,7 @@ class OpBuilderRegistrations {
const IOpBuilder* GetOpBuilder(const std::string& onnx_op_type);

void CreateArgMaxMinOpBuilder(const std::string& op_type, OpBuilderRegistrations& op_registrations);
void CreateAttentionOpBuilder(const std::string& op_type, OpBuilderRegistrations& op_registrations);
void CreateBatchNormalizationOpBuilder(const std::string& op_type, OpBuilderRegistrations& op_registrations);
void CreateBernoulliOpBuilder(const std::string& op_type, OpBuilderRegistrations& op_registrations);
void CreateCastOpBuilder(const std::string& op_type, OpBuilderRegistrations& op_registrations);
Expand Down
Loading
Loading