[llvm-branch-commits] [mlir] [mlir][linalg] Decompose winograd operators (PR #96183)

Oleksandr Alex Zinenko via llvm-branch-commits llvm-branch-commits at lists.llvm.org
Mon Jun 24 06:08:20 PDT 2024


================
@@ -323,5 +1089,12 @@ void populateWinogradConv2DPatterns(RewritePatternSet &patterns, int64_t m,
   patterns.insert<WinogradConv2DNhwcFhwc>(context, m, r);
 }
 
+void populateDecomposeWinogradOpsPatterns(RewritePatternSet &patterns) {
+  MLIRContext *context = patterns.getContext();
+  patterns.insert<DecomposeWinogradFilterTransform>(context);
+  patterns.insert<DecomposeWinogradInputTransform>(context);
+  patterns.insert<DecomposeWinogradOutputTransform>(context);
----------------
ftynse wrote:

```suggestion
  patterns.insert<DecomposeWinogradFilterTransform,
                  DecomposeWinogradInputTransform,
                  DecomposeWinogradOutputTransform>(context);
```

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


More information about the llvm-branch-commits mailing list