[Mlir-commits] [mlir] [Linalg] Add basic infra to add matchers for linalg.*conv*/*pool* ops (PR #163724)
Andrzej WarzyĆski
llvmlistbot at llvm.org
Sat Oct 25 17:26:59 PDT 2025
================
@@ -240,6 +240,475 @@ bool isReductionIterator(utils::IteratorType iteratorType) {
return iteratorType == utils::IteratorType::reduction;
}
+//===----------------------------------------------------------------------===//
+// Convolution matcher utilities
+//===----------------------------------------------------------------------===//
+
+/// Utility to match block body for linalg.pool* ops.
+template <typename... OpTypes>
+static bool bodyMatcherForPoolOps(Value yieldVal, Block *body) {
----------------
banach-space wrote:
How come we only check the body for pooling convs and neither for depthwise nor regular convs?
https://github.com/llvm/llvm-project/pull/163724
More information about the Mlir-commits
mailing list