jax
34c2dbfd
- Fix lowering code for ROCm RNN
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
257 days ago
Fix lowering code for ROCm RNN
References
ci_fix_rnn_lowering-upstream
#28036 - [ROCm] Fix lowering code for ROCm RNN
#477 - CI: 06/20/25 upstream sync
#31381 - Remove old ROCm build code
#31720 - Fix ann_test.py numerical bug in target reshape
#31768 - [ROCm] Support lowering through PJRT_Triton_Extension
#32115 - Relax version requirements for ROCm Jax Plugin wheels
#33157 - Resolve undefined behavior in bitshift unit test
#33186 - Make nvidia version data optional for ROCm builds
#579 - Create rocm-test-requirements.txt
#580 - GESVDJ support for ROCm GPUs in JAX
#581 - Fix/pallas tests shared memory
#584 - Use plain bazel to test jax, use hermetic rocm dependency
#585 - update a test for checking zero ROCm GPU event
Author
Ruturaj4
Committer
invalid-email-address
Parents
1926b99b
Loading