jax
46df13a2 - [Mosaic] NFC: Refactor tpu::PackMaskOp to take > 2 sources.

Commit

1 day ago

[Mosaic] NFC: Refactor tpu::PackMaskOp to take > 2 sources. It's essentially the vmask version of tpu::PackSubelementsOp. This can be used to support 32->8 mask packing. PiperOrigin-RevId: 879091253

References

#35283 - [ROCm] Add ROCm pytest results collection

#35636 - [Mosaic] NFC: Refactor tpu::PackMaskOp to take > 2 sources.

#35668 - [ROCm] Fix plugin init script

Author

WindQAQ

Committer

Google-ML-Automation

Parents

2e8984a9

jax 46df13a2 - [Mosaic] NFC: Refactor tpu::PackMaskOp to take > 2 sources.

jax
46df13a2 - [Mosaic] NFC: Refactor tpu::PackMaskOp to take > 2 sources.