Skip to content

[Relax][PyTorch] Add decomposed operator support for AdaptiveAvgPool #18437

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Merged
tlopex merged 2 commits into from
Nov 13, 2025
Merged

[Relax][PyTorch] Add decomposed operator support for AdaptiveAvgPool #18437

tlopex merged 2 commits into from
Nov 13, 2025

Conversation

@guan404ming
Copy link
Member

@guan404ming guan404ming commented Nov 13, 2025 •
edited
Loading

Related Issue

Why

  • When run_ep_decomposition=True is enabled, PyTorch decomposes adaptive_avg_pool operators into lower-level _adaptive_avg_pool*.default variants, which were not supported.

How

  • Added support for _adaptive_avg_pool1d.default, _adaptive_avg_pool2d.default, and _adaptive_avg_pool3d.default in exported_program_translator.py
  • Updated tests to use run_ep_decomposition=True including conv tests as well

@guan404ming
Copy link
Member Author

guan404ming commented Nov 13, 2025

cc @tlopex

@guan404ming guan404ming marked this pull request as ready for review November 13, 2025 10:34
tlopex
tlopex approved these changes Nov 13, 2025
View reviewed changes
Copy link
Member

@tlopex tlopex left a comment

Choose a reason for hiding this comment

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

LGTM! Thanks

@tlopex tlopex merged commit 506a0bb into apache:main Nov 13, 2025
14 checks passed
@guan404ming
Copy link
Member Author

guan404ming commented Nov 14, 2025

Thanks!

@guan404ming guan404ming deleted the add-decomposed-support-for-adp branch November 14, 2025 05:22
@ysh329 ysh329 mentioned this pull request Feb 1, 2026
Closed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@tlopex tlopex tlopex approved these changes

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@guan404ming @tlopex