[MLIR][XeGPU] Fix layout inference issues blocking MXFP_GEMM test (#196243)
This branch fixes layout inference issues in XeGPU passes that were
blocking MXFP (microscaled floating point) GEMM workloads:
- Fix bitcast layout adjustment to use result shape instead of source
shape. The setupBitCastResultLayout function were incorrectly bounding
the layout adjustment loop against the source shape. Added tests.
- Fix blocking pass to drop inst_data from anchor operations. Operations
whose shape already matches inst_data don't get unrolled, so their
layout attributes retained stale inst_data that broke downstream passes.
Now inst_data is unconditionally stripped from all op attributes after
blocking.
- Propagate layout to both results of vector.deinterleave. The layout
recovery pass was only setting the layout on result 0, leaving result 1
without a layout.
Test plan
- Added mlir/test/Integration/Dialect/XeGPU/WG/simple_ mxfp_gemm.mlir
integration test exercising the full
MXFP GEMM pipeline (bitcast, deinterleave, transpose, interleave,
dpas_mx).
Assisted by Claude
---------
Co-authored-by: Claude Sonnet 4.5 <noreply@anthropic.com>