We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
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
I see that there are docs for triton support but only for GPU? Is it possible for TPU to use triton?
The content you are editing has changed. Please copy your edits and refresh the page.
The text was updated successfully, but these errors were encountered:
short answer is no, we used pallas as the custom kernel language for TPU. Checkout https://pytorch.org/xla/master/features/pallas.html
Sorry, something went wrong.
No branches or pull requests
❓ Questions and Help
I see that there are docs for triton support but only for GPU? Is it possible for TPU to use triton?
Tasks
The text was updated successfully, but these errors were encountered: