[llvm-branch-commits] [mlir] [mlir][linalg] Decompose winograd operators (PR #96183)
Hsiangkai Wang via llvm-branch-commits
llvm-branch-commits at lists.llvm.org
Thu Jun 27 01:41:09 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);
----------------
Hsiangkai wrote:
Done.
https://github.com/llvm/llvm-project/pull/96183
More information about the llvm-branch-commits
mailing list