[Mlir-commits] [mlir] [MLIR][XeGPU] Add distribution pattern for convertLayoutOp (PR #184826)

Charitha Saumya llvmlistbot at llvm.org
Wed Mar 11 10:06:51 PDT 2026


================
@@ -696,6 +696,28 @@ struct SgToWiStoreScatter : public OpConversionPattern<xegpu::StoreScatterOp> {
   }
 };
 
+struct SgToWiConvertLayout
+    : public OpConversionPattern<xegpu::ConvertLayoutOp> {
+  using OpConversionPattern<xegpu::ConvertLayoutOp>::OpConversionPattern;
+
+  LogicalResult
+  matchAndRewrite(xegpu::ConvertLayoutOp op, OpAdaptor adaptor,
+                  ConversionPatternRewriter &rewriter) const override {
+    auto inputLayout = op.getInputLayoutAttr();
+    auto targetLayout = op.getTargetLayoutAttr();
+
+    if (!inputLayout || !targetLayout)
----------------
charithaintc wrote:

does the op definition allow this. otherwise could be redundant check.

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


More information about the Mlir-commits mailing list