Skip to content

Move output tensor resize to before XNNPACK subgraph execution#17661

Open
GregoryComer wants to merge 1 commit intopytorch:mainfrom
GregoryComer:export-D94148443
Open

Move output tensor resize to before XNNPACK subgraph execution#17661
GregoryComer wants to merge 1 commit intopytorch:mainfrom
GregoryComer:export-D94148443

Conversation

@GregoryComer
Copy link
Member

Summary:
ET's XNNPACK delegate currently resizes output tensors after running the XNNPACK subgraph. This is normally fine, as the buffer is large enough. However, if there is a logic bug or corrupt model file, the current code doesn't catch if XNNPACK's output is larger than the planned buffer until after execution, where it may crash.

To fix this, I've moved the resize up. This fails gracefully in the above case.

Differential Revision: D94148443

@pytorch-bot
Copy link

pytorch-bot bot commented Feb 24, 2026

🔗 Helpful Links

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

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

❌ 5 New Failures, 5 Unrelated Failures

As of commit 45ee5c1 with merge base 3319157 (image):

NEW FAILURES - The following jobs have failed:

FLAKY - The following jobs failed but were likely due to flakiness present on trunk:

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 Feb 24, 2026
@meta-codesync
Copy link
Contributor

meta-codesync bot commented Feb 24, 2026

@GregoryComer has exported this pull request. If you are a Meta employee, you can view the originating Diff in D94148443.

@github-actions
Copy link

This PR needs a release notes: label

If your change should be included in the release notes (i.e. would users of this library care about this change?), please use a label starting with release notes:. This helps us keep track and include your important work in the next release notes.

To add a label, you can comment to pytorchbot, for example
@pytorchbot label "release notes: none"

For more information, see
https://github.com/pytorch/pytorch/wiki/PyTorch-AutoLabel-Bot#why-categorize-for-release-notes-and-how-does-it-work.

GregoryComer added a commit to GregoryComer/executorch that referenced this pull request Feb 24, 2026
…ch#17661)

Summary:

ET's XNNPACK delegate currently resizes output tensors after running the XNNPACK subgraph. This is normally fine, as the buffer is large enough. However, if there is a logic bug or corrupt model file, the current code doesn't catch if XNNPACK's output is larger than the planned buffer until after execution, where it may crash.

To fix this, I've moved the resize up. This fails gracefully in the above case.

Differential Revision: D94148443
…ch#17661)

Summary:

ET's XNNPACK delegate currently resizes output tensors after running the XNNPACK subgraph. This is normally fine, as the buffer is large enough. However, if there is a logic bug or corrupt model file, the current code doesn't catch if XNNPACK's output is larger than the planned buffer until after execution, where it may crash.

To fix this, I've moved the resize up. This fails gracefully in the above case.

Differential Revision: D94148443
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported meta-exported

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant