jax
ee6208fa
- internal change
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
4 days ago
internal change PiperOrigin-RevId: 882609931
References
#26604 - [ROCm][pallas:triton] Fix atomic min/max lowering for uint and float types
#34135 - [ROCm] update to test if there are GPU events when doing profiling on…
#34971 - [ROCm] fix the performance issue when n=1 or 2
#35534 - [ROCm] bring gesdd for computing SVD on ROCm
#35611 - [ROCm] Skip ann_test test_pmap on ROCm due to IndivisibleError
#35836 - [Pallas/Mosaic GPU] Use `CustomPrimitiveOp` when lowering `semaphore_{signal,wait}` with WG semantics.
#35917 - [ROCm] Skip test_tridiagonal_solve_grad0 due to rocSparse numerical issue
#35932 - Rocm/s3 wheel downloads
#35944 - Add ROCm xdist device pinning in conftest.py
Author
Google-ML-Automation
Committer
hawkinsp
Parents
babbce66
Loading