[Mlir-commits] [mlir] [mlir][memref] Fold subview into nd-tensor descriptor (PR #88698)

Oleksandr Alex Zinenko llvmlistbot at llvm.org
Mon Apr 15 02:59:47 PDT 2024


================
@@ -323,6 +324,16 @@ class NvgpuAsyncCopyOpSubViewOpFolder final
   LogicalResult matchAndRewrite(nvgpu::DeviceAsyncCopyOp copyOp,
                                 PatternRewriter &rewriter) const override;
 };
+
+/// Merges subview operation with xegpu.create_nd_tdesc operation.
+class XegpuCreateNdDescOpSubViewOpFolder final
+    : public OpRewritePattern<xegpu::CreateNdDescOp> {
+public:
+  using OpRewritePattern<xegpu::CreateNdDescOp>::OpRewritePattern;
+
+  LogicalResult matchAndRewrite(xegpu::CreateNdDescOp descOp,
+                                PatternRewriter &rewriter) const override;
+};
----------------
ftynse wrote:

This belongs to the xegpu dialect.

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


More information about the Mlir-commits mailing list