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

Update request: jaxlib with ROCm support #302676

Open

dwf opened this issue Apr 8, 2024 · 1 comment

Labels

0.kind: packaging request

Contributor

dwf commented Apr 8, 2024 •

edited

Loading

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

Add a 👍 reaction to issues you find important.

The text was updated successfully, but these errors were encountered:

dwf added the 0.kind: packaging request label

Contributor Author

dwf commented Apr 8, 2024 •

edited

Loading

Log from attempting to use the unsuccessful build (AMD_LOG_LEVEL=2 python)
https://gist.github.com/dwf/f071422e3d3e09a4024a70c649b7cd51

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment