Skip to content

NXP backend: Add support for aten.avg_pool1d.default.#17412

Open
MartinPavella wants to merge 1 commit intopytorch:mainfrom
nxp-upstream:nxg01483/EIEX-705-add-support-for-aten.avg_pool1d-to-nxp-backend
Open

NXP backend: Add support for aten.avg_pool1d.default.#17412
MartinPavella wants to merge 1 commit intopytorch:mainfrom
nxp-upstream:nxg01483/EIEX-705-add-support-for-aten.avg_pool1d-to-nxp-backend

Conversation

@MartinPavella
Copy link
Collaborator

@MartinPavella MartinPavella commented Feb 12, 2026

Summary

During the conversion from aten dialect to edge dialect, ExecuTorch replaces the avg_pool1d with avg_pool_2d and adds an aten,unsqueeze before, to introduce the 4th dimension of size 1, and an aten.squeeze is added after to remove it. The avg_pool1d is then supported by the existing AvgPool2DConverter.

Test plan

Unit-tests provided.

cc @robert-kalmar @JakeStevens @digantdesai

@MartinPavella MartinPavella added module: nxp Issues related to NXP Neutron NPU delegation and code under backends/nxp/ release notes: nxp Changes to the NXP Neutron backend delegate labels Feb 12, 2026
@pytorch-bot
Copy link

pytorch-bot bot commented Feb 12, 2026

🔗 Helpful Links

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

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

❌ 4 New Failures

As of commit 8b2200a with merge base 331ac4a (image):

NEW FAILURES - The following jobs have failed:

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 12, 2026
@MartinPavella
Copy link
Collaborator Author

The changes are based on another PR (#16540) so only the last commit is relevant.

@MartinPavella MartinPavella force-pushed the nxg01483/EIEX-705-add-support-for-aten.avg_pool1d-to-nxp-backend branch from 7984709 to ce9a9ed Compare February 12, 2026 14:35
@MartinPavella
Copy link
Collaborator Author

@novak-vaclav can you please take a look at this PR?

@roman-janik-nxp roman-janik-nxp self-requested a review February 17, 2026 09:50
@novak-vaclav
Copy link
Contributor

Looks good to me 👍

@MartinPavella MartinPavella force-pushed the nxg01483/EIEX-705-add-support-for-aten.avg_pool1d-to-nxp-backend branch 2 times, most recently from 4d3a51e to 1bed0ab Compare February 24, 2026 13:11
@MartinPavella MartinPavella force-pushed the nxg01483/EIEX-705-add-support-for-aten.avg_pool1d-to-nxp-backend branch from 1bed0ab to 42b7291 Compare February 26, 2026 07:02
@MartinPavella MartinPavella force-pushed the nxg01483/EIEX-705-add-support-for-aten.avg_pool1d-to-nxp-backend branch from 42b7291 to 8b2200a Compare February 26, 2026 12:55
@novak-vaclav
Copy link
Contributor

Thank you for incorporating my comments into the code @MartinPavella
Everything looks good to me 👍

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. module: nxp Issues related to NXP Neutron NPU delegation and code under backends/nxp/ release notes: nxp Changes to the NXP Neutron backend delegate

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants