-
Notifications
You must be signed in to change notification settings - Fork 35
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
Pallas upstream is now working? #219
Comments
I think you might not have Triton installed? If Triton is installed properly the pallas_call should work. |
Right.. Actually, my workload needs both torch and jax, but recent torch requires triton==2.0.0 while jax-triton requires triton-nightly. There is version conflict so that I decided to use triton==2.0.0 instead of triton-nightly. Could it be possible under this setting? Currently, triton==2.0.0 is installed in my environment. |
I saw the pallas concept in official latest jax docs, and follow up the pallas quickstart section.
I installed latest jaxlib and jax using github head.
I encountered the following error.
Do I have to install something different? or has it not fully upstreamed yet?
The text was updated successfully, but these errors were encountered: