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

[Jianhui Li]

================
@@ -852,6 +853,16 @@ def XeGPU_LoadGatherOp : XeGPU_Op<"load", [MemoryEffects<[MemRead]>]> {
       return getSource().getType();
     }
 
+    xegpu::DistributeLayoutAttr getDistributeLayout() {
+      xegpu::DistributeLayoutAttr layout = nullptr;
+      if (auto tdescType = getTensorDescType()) {
----------------
Jianhui-Li wrote:

we are deprecating the load_gather w/ tdesc format, so no need to check it here.  

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