jax
b8dbb940
- [Pallas:MGPU] Fix Pallas `swap` lowering rule for scalar constants.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
37 days ago
[Pallas:MGPU] Fix Pallas `swap` lowering rule for scalar constants. PiperOrigin-RevId: 839724087
References
#33686 - [Pallas:MGPU] Fix Pallas `swap` lowering rule for scalar constants.
#34135 - [ROCm] update to test if there are GPU events when doing profiling on…
#592 - Upstream ci download jax rocm wheels
Author
allanrenucci
Committer
Google-ML-Automation
Parents
e43b6cfe
Loading
