Skip to content

[Relax][PyTorch] Add decomposed operator support for MaxPool - #18446

Merged
tlopex merged 1 commit into
apache:mainfrom
guan404ming:add-decomposed-support-for-mp
Nov 14, 2025
Merged

[Relax][PyTorch] Add decomposed operator support for MaxPool#18446
tlopex merged 1 commit into
apache:mainfrom
guan404ming:add-decomposed-support-for-mp

Conversation

@guan404ming

Copy link
Copy Markdown
Member

Related Issue

Why

  • When run_ep_decomposition=True is enabled, PyTorch decomposes max_pool operators into max_pool*_with_indices.default variants, which were not supported.

How

  • Added support for max_pool*_with_indices operators
  • Updated tests to use run_ep_decomposition=True

@guan404ming
guan404ming marked this pull request as ready for review November 14, 2025 05:21
@guan404ming

Copy link
Copy Markdown
Member Author

cc @tlopex

@tlopex tlopex left a comment

Copy link
Copy Markdown
Member

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 e5fb395 into apache:main Nov 14, 2025
14 checks passed
@guan404ming
guan404ming deleted the add-decomposed-support-for-mp branch November 14, 2025 05:28
@guan404ming

Copy link
Copy Markdown
Member Author

Thanks!

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.

2 participants