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

[onnx] Lowerings from onnx.transpose #2641

Merged
merged 3 commits into from
Dec 15, 2023
Merged

Conversation

rsuderman
Copy link
Contributor

Lowerings for transpose from ONNX to aten. Implementation depends on making multiple aten.transpose
operations swapping pairs of dimensions. As onnx.transpose can swap around any dimensions it may
require constructing multiple aten.transpose.

Started work on the `transpose` lowerings for ONNX to Torch.
Copy link
Collaborator

@dan-garvey dan-garvey left a comment

Choose a reason for hiding this comment

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

if you can fix the merge conflict and ping me I'll reapprove.

I like the use of swap here = )

@rsuderman
Copy link
Contributor Author

I should be good to land without a re-approve. Usually merges do not invalidate.

@rsuderman rsuderman merged commit 705ea95 into llvm:main Dec 15, 2023
5 checks passed
@rsuderman rsuderman deleted the Transpose branch February 28, 2024 20:41
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