Skip to content
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.

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

[Stablehlo] Improve the lowering of pool op in stablehlo #3259

Merged
merged 4 commits into from
Apr 30, 2024

Conversation

Xinyu302
Copy link
Contributor

  1. Handle case stride == None
  2. add avgpool3d maxpool1d maxpool3d lowering

Copy link
Collaborator

@qingyunqu qingyunqu left a comment

Choose a reason for hiding this comment

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

nit: add some testcases for max_pool1d

@qingyunqu qingyunqu merged commit f32ada9 into llvm:main Apr 30, 2024
3 checks passed
archana-ramalingam pushed a commit to archana-ramalingam/torch-mlir that referenced this pull request May 8, 2024
1. Handle case stride == None
2. add avgpool3d maxpool1d  maxpool3d lowering
josel-amd added a commit to Xilinx/torch-mlir that referenced this pull request Jun 7, 2024
Support for the operator in HLO has been implemented in

llvm#3259

but that change is not in this fork yet.
josel-amd added a commit to Xilinx/torch-mlir that referenced this pull request Jun 10, 2024
Support for the operator in HLO has been implemented in

llvm#3259

but that change is not in this fork yet.
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