jax
5bd8a46e - [Pallas:MGPU] Make sure that the warp axis returns indices in [0, 3)
Commit
View On GitHub
Commit
15 days ago
[Pallas:MGPU] Make sure that the warp axis returns indices in [0, 3) PiperOrigin-RevId: 895859912
Author
apaszke apaszke
Parents
e0696314
FAQ Terms Privacy Refunds Impressum
Loading