jax
81cb1add - [MosaicGPU] Add a repr to FragmentedArray so pallas:mosaic_gpu errors are more readable.

Commit

1 year ago

[MosaicGPU] Add a __repr__ to FragmentedArray so pallas:mosaic_gpu errors are more readable. PiperOrigin-RevId: 655082250

References

Author

cperivol

Committer

a-googler

Parents

jax 81cb1add - [MosaicGPU] Add a __repr__ to FragmentedArray so pallas:mosaic_gpu errors are more readable.