[Mlir-commits] [mlir] [MLIR][XeGPU] Fix layout inference issues blocking MXFP_GEMM test (PR #196243)

[Sang Ik Lee]

================
@@ -1003,6 +1006,7 @@ xegpu::DistributeLayoutAttr xegpu::setupBitCastResultLayout(
   int resElemTyBitWidth = resVecTy.getElementType().getIntOrFloatBitWidth();
 
   ArrayRef<int64_t> srcShape = srcVecTy.getShape();
+  ArrayRef<int64_t> resShape = resVecTy.getShape();
----------------
silee2 wrote:

Many places that used `srcShape` now uses `resShape` which doesn't look like a trivial change.
Can you add a unit test that tests the updated logic?

https://github.com/llvm/llvm-project/pull/196243