jax
259194a6 - [Pallas] Fix shard_axis in dma_start interpret mode rule.

Commit

1 year ago

[Pallas] Fix shard_axis in dma_start interpret mode rule. PiperOrigin-RevId: 703192497

References

Author

justinjfu

Committer

hawkinsp

Parents