[Mlir-commits] [mlir] [MLIR][XeGPU] Propagate layout onto loop-carried iter_arg entry edges (PR #198862)

[Jianhui Li]

================
@@ -1836,6 +1836,10 @@ LogicalResult xegpu::propagateLayouts(OpBuilder &builder, Operation *target,
             r = updateControlFlowOps(builder, branchTermOp,
                                      getXeGPULayoutForValue);
           })
+          .Case([&](mlir::RegionBranchOpInterface branchOp) {
+            r = xegpu::propagateRegionArgsToInits(branchOp,
+                                                  getXeGPULayoutForValue);
----------------
Jianhui-Li wrote:

Could you please change the name of getXeGPULayoutForValue() to something like getLayoutFromPropagation()? Since the function getsLayoutInfo and layout atttributes as a result of propagation. 

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