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] Init MHLO linear op patterns #1132

Merged
merged 4 commits into from Aug 4, 2022

Conversation

tanyokwok
Copy link
Collaborator

See RFC #999

Co-authored-by: Bairen Yi yibairen.byron@bytedance.com
Co-authored-by: Jiawei Wu xremold@gmail.com
Co-authored-by: Tianyou Guo tianyou.gty@alibaba-inc.com
Co-authored-by: Xu Yan yancey.yx@alibaba-inc.com
Co-authored-by: Ziheng Jiang ziheng.jiang@bytedance.com

@tanyokwok
Copy link
Collaborator Author

Ready for review @silvasean @ZihengJiang @Vremold @Yancey1989

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

lib/Conversion/TorchToMhlo/MatmulOp.cpp Outdated Show resolved Hide resolved
Copy link
Collaborator

@ZihengJiang ZihengJiang left a comment

Choose a reason for hiding this comment

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

Should we rename this part to Linear.cpp to align with https://github.com/llvm/torch-mlir/blob/main/lib/Conversion/TorchToLinalg/Linear.cpp

Maybe let's remove all the "Op" suffix in the file names

Copy link
Collaborator

@Yancey1989 Yancey1989 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.

Great!

@ZihengJiang ZihengJiang changed the title [MHLO] Init MHLO matmul op patterns [MHLO] Init MHLO linear op patterns Aug 3, 2022
@ZihengJiang
Copy link
Collaborator

Could you also add the file into the bazel build like this? #1148

Tanyo Kwok added 4 commits August 4, 2022 09:26
See RFC llvm#999

Co-authored-by: Bairen Yi yibairen.byron@bytedance.com
Co-authored-by: Jiawei Wu xremold@gmail.com
Co-authored-by: Tianyou Guo tianyou.gty@alibaba-inc.com
Co-authored-by: Xu Yan yancey.yx@alibaba-inc.com
Co-authored-by: Ziheng Jiang ziheng.jiang@bytedance.com
@tanyokwok
Copy link
Collaborator Author

Could you also add the file into the bazel build like this? #1148

Thanks! Done.

@ZihengJiang ZihengJiang merged commit f0a24f5 into llvm:main Aug 4, 2022
@ZihengJiang
Copy link
Collaborator

Merged. Thanks @fortianyou

@tanyokwok tanyokwok deleted the upstream-06-mhlo-matmul branch August 4, 2022 02:23
qedawkins pushed a commit to nod-ai/torch-mlir that referenced this pull request Oct 3, 2022
* bump llvm-project to 881ff4e4ebe8cc0cc045c7c167cffb01f94f27f8

Signed-off-by: Liam Fitzpatrick <liamf@xilinx.com>
Co-authored-by: Alexandre Eichenberger <alexe@us.ibm.com>
Co-authored-by: gongsu832 <gong_su@hotmail.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.

None yet

4 participants