-
Notifications
You must be signed in to change notification settings - Fork 2
Add /opt/rocm/lib to RPATH so missing ROCm libs are found #147
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
Conversation
gulsumgudukbay
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Could you please add how you tested this? Do the tests work now?
Yes, I can start the nightly wheel builds to make sure the issue is gone. |
Co-authored-by: Gulsum Gudukbay Akbulut <[email protected]>
f3b635d to
76dc99e
Compare
|
These wheels generated in: https://github.com/ROCm/rocm-jax/actions/runs/18574505896, will be tested. |
|
Started another nightly: https://github.com/ROCm/rocm-jax/actions/runs/18576729511, cancelled the previous one. |
|
I previously hit the same issue on this branch, in a .venv and with prebuilt wheels. I have not been able to re-run tests yet, but the earlier JAX 0.6.0 PR and prior experience, this should address the problem. I 'll follow up with a verification run when I have cycles. |
See PR #135
Similarly, the wheels generated in nightly builds had the same issue while running a workload.