[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