Skip to content

[NPUW]Annotate each Add in SDPA to make per-ATTN-subgraph mask-skipping decisions - #37407

Open
intelgaoxiong wants to merge 1 commit into
openvinotoolkit:masterfrom
intelgaoxiong:xiong/per_layer_annotate
Open

[NPUW]Annotate each Add in SDPA to make per-ATTN-subgraph mask-skipping decisions#37407
intelgaoxiong wants to merge 1 commit into
openvinotoolkit:masterfrom
intelgaoxiong:xiong/per_layer_annotate

Conversation

@intelgaoxiong

@intelgaoxiong intelgaoxiong commented Aug 13, 2026

Copy link
Copy Markdown
Contributor

Details:

The PR introduced a new pass for per-layer annotation and mask type analysis.
The per-layer mask type information will be used by model with hybrid attention for:

  1. Making per-ATTN-subgraph mask-skipping decisions
  2. "REAL" SWA transformation in the future work by @GuoliangShiIntel

Tickets:

AI Assistance:

  • AI assistance used: no / yes
  • If yes, summarize how AI was used and what human validation was performed (build/tests/manual checks).

@github-actions github-actions Bot added category: NPU OpenVINO NPU plugin category: NPUW NPUW plugin labels Aug 13, 2026
@intelgaoxiong
intelgaoxiong marked this pull request as ready for review August 13, 2026 02:07
@intelgaoxiong
intelgaoxiong requested review from a team as code owners August 13, 2026 02:07
@intelgaoxiong intelgaoxiong changed the title Annotate each Add in SDPA to make per-ATTN-subgraph mask-skipping decisions [NPUW]Annotate each Add in SDPA to make per-ATTN-subgraph mask-skipping decisions Aug 13, 2026
@intelgaoxiong
intelgaoxiong force-pushed the xiong/per_layer_annotate branch from 6e5efcf to 5933eaf Compare August 13, 2026 08:33
@intelgaoxiong

Copy link
Copy Markdown
Contributor Author

@dmatveev @kkoryun @AlexanderKalistratov Could you please review the PR? Thanks!

…isions.

Add interface to get per-SDPA's mask type.

Add unit test.

Signed-off-by: intelgaoxiong <xiong.gao@intel.com>
@intelgaoxiong
intelgaoxiong marked this pull request as draft August 14, 2026 11:13
@intelgaoxiong
intelgaoxiong marked this pull request as ready for review August 14, 2026 11:52
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

category: NPU OpenVINO NPU plugin category: NPUW NPUW plugin

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant