jax
0fd08213
- [Pallas TPU] Add flag to enable using registers to keep track of slot info
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
218 days ago
[Pallas TPU] Add flag to enable using registers to keep track of slot info PiperOrigin-RevId: 772527011
References
#29503 - [Pallas TPU] Add flag to enable using registers to keep track of slot info
#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
#579 - Create rocm-test-requirements.txt
#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…
#591 - Adding run_pytest_rocm.sh
#592 - Upstream ci download jax rocm wheels
#34304 - [ROCm] Add ROCm platform support for psend/precv collective operations
#599 - Use xla from rocm-jaxlib-v0.8.0 release branch
#34450 - [ROCm] Upstream ROCm CI Nightly Wheel Testing
#611 - Enable test_comparison_scalar tests for ROCm GPUs
#34470 - Enabled testTridiagonal for ROCm devices.
#34474 - Added support for GESVDJ on ROCm devices
#34494 - Enabled ROCm devices to default to Jacobi SVD on smaller matrices.
#632 - Add script for running bazel tests on ROCm
#634 - Skip tridiagonal solve tests on ROCm
Author
Google-ML-Automation
Committer
Google-ML-Automation
Parents
332aa354
Loading