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

[MHLO]fix lowering failed on reduction op with i32 shape #1185

Merged
merged 2 commits into from
Aug 9, 2022

Conversation

Yancey1989
Copy link
Collaborator

This PR fixed lowering failed on ten::max_dim with i32 shape type.

Copy link
Collaborator

@tanyokwok tanyokwok left a comment

Choose a reason for hiding this comment

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

LGTM

Copy link
Collaborator

@Vremold Vremold left a comment

Choose a reason for hiding this comment

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

Overall LGTM. But could we add a unit test case with i32 shape?

lib/Conversion/TorchToMhlo/Reduction.cpp Outdated Show resolved Hide resolved
@Yancey1989 Yancey1989 merged commit f83a905 into llvm:main Aug 9, 2022
@Yancey1989 Yancey1989 deleted the fix_reduction_i32_shape branch August 9, 2022 09:02
qedawkins pushed a commit to nod-ai/torch-mlir that referenced this pull request Oct 3, 2022
- install onnx into lib instead of lib64
- fix time zone setup in container
- download jsoniter jar from maven repo
- run check-mlir on s390x but ignore error for now
- run check-onnx-backend-signature
- unshallow and rename origin to upstream for onnx-mlir-dev
  image to be published

Signed-off-by: Gong Su <gong_su@hotmail.com>

Co-authored-by: Alexandre Eichenberger <alexe@us.ibm.com>
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