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

[Jianhui Li]

================
@@ -993,6 +1017,16 @@ def XeGPU_StoreScatterOp : XeGPU_Op<"store", [MemoryEffects<[MemWrite]>]> {
       return TypedValue<xegpu::TensorDescType>();
     }
 
+    xegpu::DistributeLayoutAttr getDistributeLayout() {
+      xegpu::DistributeLayoutAttr layout = nullptr;
+      if (auto tdescType = getTensorDescType()) {
----------------
Jianhui-Li wrote:

no need to support the tdesc form. 

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