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
Technically JAX [and TensorFlow] support ROCm via XLA.
I've taken a run at this in https://github.com/dwf/nixpkgs/tree/jax_rocm and with a pile of hacks have gotten it to build but it fails to run a matrix multiply correctly. I used rocmPackages_5 due to #299156 blocking 6.
rocmPackages_5
Previous work on TensorFlow at Madouura@344aa78
cc #197885
Add a 👍 reaction to issues you find important.
The text was updated successfully, but these errors were encountered:
Log from attempting to use the unsuccessful build (AMD_LOG_LEVEL=2 python) https://gist.github.com/dwf/f071422e3d3e09a4024a70c649b7cd51
AMD_LOG_LEVEL=2 python
Sorry, something went wrong.
No branches or pull requests
Technically JAX [and TensorFlow] support ROCm via XLA.
I've taken a run at this in https://github.com/dwf/nixpkgs/tree/jax_rocm and with a pile of hacks have gotten it to build but it fails to run a matrix multiply correctly. I used
rocmPackages_5
due to #299156 blocking 6.Previous work on TensorFlow at Madouura@344aa78
cc #197885
Add a 👍 reaction to issues you find important.
The text was updated successfully, but these errors were encountered: