jax
58fb923f
- Automated g4 rollback of changelist 784947368.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
167 days ago
Automated g4 rollback of changelist 784947368. Reverts 44b6204b4b110145000be12a76bb1224b0c7406f PiperOrigin-RevId: 785486706
References
#30353 - Automated g4 rollback of changelist 784947368.
#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
#34135 - [ROCm] update to test if there are GPU events when doing profiling on…
Author
Google-ML-Automation
Committer
Google-ML-Automation
Parents
b97ad423
Loading