Skip to content

[TOSA] Add upsample_nearest2d, split_dim, outer, GELU tanh mode and misc #3886

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

Merged
merged 1 commit into from
Dec 2, 2024

Conversation

justin-ngo-arm
Copy link
Contributor

  • Add Torch to TOSA lowering for the following ops:
    • torch.aten.upsample_nearest2d
    • torch.aten.upsample_nearest2d.vec
    • torch.aten.outer
    • torch.prims.split_dim
  • Add Tanh approximation mode for GELU lowering
  • Add different types support for compare ops
  • Add different input and output types support for linalg vector norm lowering
  • Update xfail with new e2e results
  • Add new LIT tests to basic.mlir

Change-Id: I7b1d44d94319cf94fcc9d234cc07708ef9ce321e

@sjarus sjarus requested a review from sahas3 November 21, 2024 19:15
- Add Torch to TOSA lowering for the following ops:
  + torch.aten.upsample_nearest2d
  + torch.aten.upsample_nearest2d.vec
  + torch.aten.outer
  + torch.prims.split_dim
- Add Tanh approximation mode for GELU lowering
- Add different types support for compare ops
- Add different input and output types support for linalg vector norm
  lowering
- Update xfail with new e2e results
- Add new LIT tests to basic.mlir

Signed-off-by: Justin Ngo <justin.ngo@arm.com>
Change-Id: I7b1d44d94319cf94fcc9d234cc07708ef9ce321e
@sjarus sjarus merged commit 8711d3e into llvm:main Dec 2, 2024
3 checks passed
@justin-ngo-arm justin-ngo-arm deleted the dev branch December 2, 2024 18:17
rahuls-cerebras added a commit that referenced this pull request Jan 3, 2025
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