Skip to content

Fix EP assignment reporting after NHWC layout transformation - #33184

Merged
mirounga merged 1 commit into
microsoft:mainfrom
saiyambharara:fix/32998-nhwc-assignment-info
Oct 8, 2026
Merged

mirounga merged 1 commit into
microsoft:mainfrom
saiyambharara:fix/32998-nhwc-assignment-info

Conversation

@saiyambharara

@saiyambharara Saiyam Bharara (saiyambharara) commented Oct 7, 2026 •

Copy link
Copy Markdown
Contributor

Description

NHWC layout transformation provisionally assigns nodes before final capability processing. The singleton admission check rejects these already-tagged nodes, omitting assignment records despite successful execution.

Track provisional assignments from the current layout pass and admit their singleton capabilities. Continue rejecting assignments finalized in earlier rounds to prevent duplicate records after function inlining.

Add regression coverage for unchanged singleton nodes, repeated partitioning in both layouts, and reconstructed NHWC Conv nodes.

Fixes #32998.

Validation

  • Windows Release: 100 selected partitioning/plugin tests passed.
  • Real WebGPU execution: original reproduction, Add chain, mixed CPU/GPU placement, and all four layout/inlining configurations passed.
  • clang-format and git diff --check passed.

Track provisional assignments from the current NHWC layout pass so singleton capabilities produce assignment records without reporting previously finalized nodes again.
Copilot AI balanced review requested due to automatic review settings October 7, 2026 22:58

Copilot AI left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Copilot was unable to review this pull request because the user who requested the review has reached their quota limit.

@azure-pipelines

Copy link
Copy Markdown
Azure Pipelines:
There may be pipelines that require an authorized user to comment /azp run to run.

@saiyambharara

Copy link
Copy Markdown
Contributor Author

Edward Chen (@edgchen1) Tianlei Wu (@tianleiwu) — could you review this fix for #32998? It tracks provisional assignments from the current NHWC pass so singleton assignment records are emitted without duplicating records when function inlining reruns partitioning. Regression coverage includes both layouts, both inlining modes, and reconstructed NHWC Conv nodes.

Copilot AI left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

🟢 Approval recommended

The targeted fix correctly distinguishes current layout-pass assignments and includes coverage for transformed and repeatedly partitioned graphs.

0 open findings

🧠 Review effort: Balanced


Give feedback about Copilot approvals in this survey to enter a drawing for a $150 gift card.

@mirounga
mirounga merged commit 430f563 into microsoft:main Oct 8, 2026
This was referenced Oct 9, 2026
This was referenced Oct 14, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[Plugin EP] Graph assignment API returns no subgraphs for an executed Add

3 participants