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

Implement torch.aten.diag_embed #288

Closed
Tracked by #347 ...
schnkmwt opened this issue Dec 21, 2023 · 2 comments
Closed
Tracked by #347 ...

Implement torch.aten.diag_embed #288

schnkmwt opened this issue Dec 21, 2023 · 2 comments

Comments

@schnkmwt
Copy link

schnkmwt commented Dec 21, 2023

This issues tracks implementation of torch.aten.diag_embed from the list of un-supported torch-mlir ops in:
#119 | #215

@schnkmwt
Copy link
Author

schnkmwt commented Feb 7, 2024

Added PR: llvm/torch-mlir#2885
Please help review!

newling pushed a commit to llvm/torch-mlir that referenced this issue Mar 22, 2024
This PR adds lowering of diag_embed to linalg dilect.
Tracked in nod-ai/SHARK-ModelDev#288

---------

Co-authored-by: sachink <[email protected]>
@schnkmwt
Copy link
Author

PR Merged!

@schnkmwt schnkmwt removed their assignment Mar 28, 2024
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

No branches or pull requests

1 participant