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

AWS Neuron PyTorch XLA Changes #5465

Open
aws-kingrj opened this issue Aug 18, 2023 · 3 comments
Open

AWS Neuron PyTorch XLA Changes #5465

aws-kingrj opened this issue Aug 18, 2023 · 3 comments

Comments

@aws-kingrj
Copy link
Collaborator

❓ Questions and Help

Previously, AWS Neuron has made specific changes to only the torch_xla wheel that we release. We plan to move all the changes to public torch_xla so they can be released as a part of the public torch_xla wheels.

Some changes will include:

  • Support for glu operator
  • Support for count non-zero operator
  • Lowering bug fixes for bad stride info
  • Other bugs found by Neuron code base
@JackCaoG
Copy link
Collaborator

Looking forward to the PR!

@aws-kingrj
Copy link
Collaborator Author

aws-kingrj commented Aug 18, 2023 •
edited
Loading

First PR for glu operator support: #5471

@aws-kingrj
Copy link
Collaborator Author

aws-kingrj commented Aug 18, 2023 •
edited
Loading

PR for lowering bug: #5486

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@JackCaoG @aws-kingrj