Skip to content

Conversation

@guan404ming
Copy link
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
Member Author

cc @tlopex

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 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
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