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

e2e support aten.linalg_norm to aten.linalg_vector_norm #2953

Merged
merged 5 commits into from
Mar 6, 2024

Conversation

zezhang
Copy link
Collaborator

@zezhang zezhang commented Feb 27, 2024

Add e2d support for aten.linalg_norm by decompose it to aten.linalg_vector_norm.

Lowering to aten.linalg_matrix_norm is still unsupported.

To Test:

python -m e2e_testing.main -v

Copy link
Collaborator

@navahgar navahgar left a comment

Choose a reason for hiding this comment

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

LGTM. I will let @ramiro050 approve this to ensure this decomposition is okay.

Copy link
Collaborator

@ramiro050 ramiro050 left a comment

Choose a reason for hiding this comment

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

Looks mostly good. Just a few small comments

lib/Dialect/Torch/Transforms/DecomposeComplexOps.cpp Outdated Show resolved Hide resolved
lib/Dialect/Torch/Transforms/DecomposeComplexOps.cpp Outdated Show resolved Hide resolved
test/Dialect/Torch/decompose-complex-ops.mlir Outdated Show resolved Hide resolved
@zezhang zezhang force-pushed the zezhang/aten_linalg_norm branch from 56853f8 to f881f41 Compare February 29, 2024 04:14
@zezhang zezhang requested a review from ramiro050 February 29, 2024 04:14
@zezhang
Copy link
Collaborator Author

zezhang commented Mar 5, 2024

@ramiro050 If you could take another look before merging.

Copy link
Collaborator

@ramiro050 ramiro050 left a comment

Choose a reason for hiding this comment

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

LGTM! Thanks

@zezhang zezhang merged commit aa7c9a9 into llvm:main Mar 6, 2024
3 checks passed
@zezhang zezhang deleted the zezhang/aten_linalg_norm branch March 6, 2024 00:31
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