[Mlir-commits] [mlir] [mlir][XeGPU] Add optional layout attribute to LoadGather StoreScatter ops (PR #163414)

[Dmitry Chigarev]

================
@@ -843,7 +843,8 @@ def XeGPU_LoadGatherOp : XeGPU_Op<"load", [MemoryEffects<[MemRead]>]> {
       AnyTypeOf<[XeGPU_MaskType, I1]>:$mask, OptionalAttr<I64Attr>:$chunk_size,
       OptionalAttr<XeGPU_CacheHintAttr>:$l1_hint,
       OptionalAttr<XeGPU_CacheHintAttr>:$l2_hint,
-      OptionalAttr<XeGPU_CacheHintAttr>:$l3_hint);
+      OptionalAttr<XeGPU_CacheHintAttr>:$l3_hint,
+      OptionalAttr<DistributeLayoutAttr>:$layout);
----------------
dchigarev wrote:

fixed,

btw [`Load/StoreMatrixOp`s take `DistributeLayoutAttr`](https://github.com/llvm/llvm-project/blob/33609bdd589d00b2bb26b9a1768a5d3c4271e22d/mlir/include/mlir/Dialect/XeGPU/IR/XeGPUOps.td#L1306), I believe we should adjust them as well eventually

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