[Mlir-commits] [mlir] [MLIR][XeGPU] Enhance Layout Propagation for broadcasting both leading dimensions and inner unit dimensions (PR #185583)

[Jianhui Li]

================
@@ -677,10 +677,17 @@ void LayoutInfoPropagation::visitVectorBroadCastOp(
   auto resShape = resultTy.getShape();
 
   size_t dimDiff = resultTy.getRank() - sourceTy.getRank();
-  for (size_t i = 0; i < srcShape.size(); i++)
-    if ((srcShape[i] == 1) && (resShape[i + dimDiff] != 1))
-      broadcast.emitWarning("broadcast must either from low-rank or same-rank "
-                            "with unit-dim, mixed scenario is not supported!");
+  if (dimDiff == 0) {
+    bool hasUnitDim =
----------------
Jianhui-Li wrote:

thank you!

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