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

[AutoBump] Merge with fixes of 11cd7cd9 (23) #254

Merged
merged 3 commits into from
Aug 23, 2024

Conversation

mgehre-amd
Copy link
Collaborator

No description provided.

zezhang and others added 3 commits May 2, 2024 00:03
…l2dWithIndicesOp (llvm#3272)

While playing with TorchDynamo on ResNet18. I notice following issues:

- `prims.convert_element_type` can’t be canonicalized even if the input
and the output share the same type

- `aten.max_pool2d_with_indices` is always used instead of
`aten.max_pool2d`, even if the second returned output (indices) has no
user

This PR fixes above issues by adding a folder to the
PrimsConvertElementTypeOp and a canonicalizer to the
AtenMaxPool2dWithIndicesOp


Lit test:

`cmake --build build --target check-torch-mlir-all`

---------

Co-authored-by: Ze Zhang <[email protected]>
Base automatically changed from bump_to_8c48135a to feature/backport_ea1_ops August 23, 2024 16:44
An error occurred while trying to automatically change base from bump_to_8c48135a to feature/backport_ea1_ops August 23, 2024 16:44
@mgehre-amd mgehre-amd merged commit 539e6b5 into feature/backport_ea1_ops Aug 23, 2024
4 checks passed
@mgehre-amd mgehre-amd deleted the bump_to_11cd7cd9 branch August 23, 2024 16:44
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.

3 participants