[Mlir-commits] [mlir] [mlir][xegpu] Add SIMT distribution support for GEMM transpose B case. (PR #155517)

[Adam Siemieniuk]

================
@@ -1276,6 +1305,142 @@ struct VectorShapeCastDistribution : public gpu::WarpDistributionPattern {
   }
 };
 
+/// Sink a memref::ExtractAlignedPointerAsIndex op feeding into yield op of an
+/// enclosing `gpu.warp_execute_on_lane_0` region. This will simply move the op
+/// outside of the warp op.
+struct MemrefExtractAlignedPointerAsIndexDistribution final
----------------
adam-smnk wrote:

Just a note for future, it can stay here for now.
This looks like pretty generic pattern, it could be moved to a different place to make it more accessible.

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