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] bert-tiny and resnet18 e2e example from torchscript to mhlo #1266

Merged
merged 1 commit into from
Aug 23, 2022

Conversation

Vremold
Copy link
Collaborator

@Vremold Vremold commented Aug 23, 2022

See RFC #999
This PR adds bert-tiny and resnet18 e2e examples from torchscript to mhlo.
Co-authored-by: Bairen Yi [email protected]
Co-authored-by: Jiawei Wu [email protected]
Co-authored-by: Tianyou Guo [email protected]
Co-authored-by: Xu Yan [email protected]
Co-authored-by: Ziheng Jiang [email protected]

@Vremold
Copy link
Collaborator Author

Vremold commented Aug 23, 2022

Since MHLO e2e test PR #1223 is not ready, and hlo-legalize-to-linalg lacks support for mhlo.dynamic_gather and mhlo.batch_norm ops, this PR just provides e2e conversion examples from torchscript to mhlo. After all these ready, we'd like to provide truely e2e examples.

Co-authored-by: Bairen Yi <[email protected]>
Co-authored-by: Jiawei Wu <[email protected]>
Co-authored-by: Tianyou Guo <[email protected]>
Co-authored-by: Xu Yan <[email protected]>
Co-authored-by: Ziheng Jiang <[email protected]>
@tanyokwok
Copy link
Collaborator

FYI, the end-to-end unit tests had been added for MHLO, #1223

@ZihengJiang ZihengJiang merged commit 1106b9a into llvm:main Aug 23, 2022
@ZihengJiang
Copy link
Collaborator

Merged. Thanks!

qedawkins pushed a commit to nod-ai/torch-mlir that referenced this pull request Oct 3, 2022
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.

4 participants