jax
f7fb9d27 - [Mosaic GPU] Simulate commuting transforms in WG semantics.

Commit
160 days ago
[Mosaic GPU] Simulate commuting transforms in WG semantics. If we cannot commute reshape we will not be able to infer the proper layout. Upon handling transforms, we try to commute them first in order to fail with a more specific error message. PiperOrigin-RevId: 923366187
Parents
Loading