jax
a5717b2f - [Mosaic GPU] Add `WarpMapOp` to the dialect.

Commit

223 days ago

[Mosaic GPU] Add `WarpMapOp` to the dialect. The op will be used as a lowering target for `pallas.core_map`. PiperOrigin-RevId: 886102626

References

Author

allanrenucci

Committer

Google-ML-Automation

Parents