jax
259194a6
- [Pallas] Fix shard_axis in dma_start interpret mode rule.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
[Pallas] Fix shard_axis in dma_start interpret mode rule. PiperOrigin-RevId: 703192497
References
#25296 - Postrelease
#176 - CI: 12/10/24 upstream sync
Author
justinjfu
Committer
hawkinsp
Parents
7e6620a5
Loading