jax
a3542f88 - [Mosaic GPU] Reconcile the swizzle of the a and b operands for wgmma in the Mosaic GPU dialect.

Commit

1 year ago

[Mosaic GPU] Reconcile the swizzle of the a and b operands for wgmma in the Mosaic GPU dialect. These need to match but we didn't ensure this previously. PiperOrigin-RevId: 771047725

References

#29437 - [Mosaic GPU] Reconcile the swizzle of the a and b operands for wgmma in the Mosaic GPU dialect.

Author

dimitar-asenov

Committer

Google-ML-Automation

Parents

9b45aacb

jax a3542f88 - [Mosaic GPU] Reconcile the swizzle of the a and b operands for wgmma in the Mosaic GPU dialect.

jax
a3542f88 - [Mosaic GPU] Reconcile the swizzle of the a and b operands for wgmma in the Mosaic GPU dialect.