[Mlir-commits] [mlir] [MLIR][XeGPU] Add distribution pattern for xegpu.load & store for sg to wi pass (PR #181917)

[Jianhui Li]

================
@@ -395,6 +395,71 @@ struct SgToWiPrefetchNd : public OpConversionPattern<xegpu::PrefetchNdOp> {
   }
 };
 
+/// Distributes a subgroup-level LoadGather (xegpu.load) op to workitem-level.
----------------
Jianhui-Li wrote:

could you please add exmples here for three use case: 1d, 2d with chunk size, 3d with leading unit dim?  The comment in warp op distribution pattern could be helpful reference. 

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