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.

Sign up for GitHub

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

Jump to bottom

torch.aten.trace #333

Closed
Tracked by #347
frafranz opened this issue Jan 10, 2024 · 3 comments
Closed
Tracked by #347

torch.aten.trace #333

frafranz opened this issue Jan 10, 2024 · 3 comments
Assignees
@frafranz

Comments

@frafranz
Copy link

No description provided.

@frafranz frafranz mentioned this issue Jan 10, 2024
Open
@frafranz frafranz self-assigned this Jan 10, 2024
@frafranz
Copy link
Author

frafranz commented Jan 10, 2024 •
edited
Loading

I am working on a lowering of torch.ops.aten.trace into linalg, by decomposition into einsum, or alternatively into diagonal and then sum.

@frafranz
Copy link
Author

frafranz commented Jan 10, 2024 •
edited
Loading

For the full list of ops being implemented, see #215.

@vivekkhandelwal1 vivekkhandelwal1 mentioned this issue Jan 12, 2024
Open
80 tasks
@frafranz frafranz reopened this Jan 23, 2024
@frafranz frafranz mentioned this issue Jan 23, 2024
Merged
newling pushed a commit to llvm/torch-mlir that referenced this issue Feb 9, 2024
@frafranz @FranzHaniel
Implement trace (#2790)
4cc62ae 
The lowering decomposes AtenTraceOp into an AtenDiagonalOp followed by
AtenSumOp.

The progress is tracked in
nod-ai/SHARK-ModelDev#333.

---------

Co-authored-by: Franz Haniel <franz.haniel@amd.com>
aartbik pushed a commit to aartbik/torch-mlir that referenced this issue Feb 9, 2024
@frafranz @FranzHaniel @aartbik
Implement trace (llvm#2790)
1d33f3f 
The lowering decomposes AtenTraceOp into an AtenDiagonalOp followed by
AtenSumOp.

The progress is tracked in
nod-ai/SHARK-ModelDev#333.

---------

Co-authored-by: Franz Haniel <franz.haniel@amd.com>
@frafranz
Copy link
Author

This has now been merges as part of llvm/torch-mlir#2790

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@frafranz frafranz

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@frafranz