jax
5bd8a46e
- [Pallas:MGPU] Make sure that the warp axis returns indices in [0, 3)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
15 days ago
[Pallas:MGPU] Make sure that the warp axis returns indices in [0, 3) PiperOrigin-RevId: 895859912
References
#36557 - [Pallas:MGPU] Make sure that the warp axis returns indices in [0, 3)
Author
apaszke
Committer
Google-ML-Automation
Parents
e0696314
Loading