[Mlir-commits] [mlir] [mlir][linalg] Add NHWC + FHWC Img2Col (PR #68708)
Quinn Dawkins
llvmlistbot at llvm.org
Wed Oct 11 10:56:03 PDT 2023
================
@@ -534,6 +668,19 @@ class ConvertConv2DNchwFchw final
return success();
}
};
+
+class ConvertConv2DNhwcFhwc final
+ : public OpRewritePattern<linalg::Conv2DNhwcFhwcOp> {
+public:
+ using OpRewritePattern::OpRewritePattern;
+
+ LogicalResult matchAndRewrite(linalg::Conv2DNhwcFhwcOp convOp,
+ PatternRewriter &rewriter) const override {
+ if (failed(rewriteInIm2Col(rewriter, convOp)))
+ return failure();
+ return success();
+ }
+};
} // end anonymous namespace
void populateConvertConv2DToImg2ColPatterns(RewritePatternSet &patterns) {
----------------
qedawkins wrote:
Add the new conversion pattern to the pattern population as well.
https://github.com/llvm/llvm-project/pull/68708
More information about the Mlir-commits
mailing list