jax
1d77720e - [Pallas] Add initial DMA interpret mode rules. Currently this only supports LOGICAL device ids with one sharding axis.
Commit
View On GitHub
Commit
1 year ago
[Pallas] Add initial DMA interpret mode rules. Currently this only supports LOGICAL device ids with one sharding axis. PiperOrigin-RevId: 644171210
Author
justinjfu justinjfu
Committer
a-googler a-googler
Parents
9556a29c
FAQ Terms Privacy Refunds Impressum
Loading