llvm-project
fb20976d - [MLIR][XeGPU] Fix layout inference issues blocking MXFP_GEMM test (#196243)

Commit
1 day ago
[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>
Author
Parents
Loading