[XLA:GPU] Move layout_assignment before fusion
This will allow code simplification and opens up new optimization. Currently we don't emit layouts inside of fusion and tracing layouts through fusion is very hard. Changing the pipeline sidesteps this issue. This is mostly perf-neutral. PiperOrigin-RevId: 188158481
Loading
Please sign in to comment