[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