jax
b6996cbd
- [Pallas][Mosaic GPU] Use NDLoopInfo as the argument to the loop body of plgpu.nd_loop
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
143 days ago
[Pallas][Mosaic GPU] Use NDLoopInfo as the argument to the loop body of plgpu.nd_loop PiperOrigin-RevId: 811413884
References
#31941 - [Pallas][Mosaic GPU] Use NDLoopInfo as the argument to the loop body of plgpu.nd_loop
Author
justinjfu
Committer
Google-ML-Automation
Parents
85edead0
Loading