[Mlir-commits] [mlir] [mlir][linalg] Implement Winograd Conv2D. (PR #94470)

Oleksandr Alex Zinenko llvmlistbot at llvm.org
Fri Jun 14 11:37:30 PDT 2024


================
@@ -154,4 +154,96 @@ def Linalg_SoftmaxOp : Linalg_Op<"softmax",
   let hasVerifier = 1;
 }
 
+def Linalg_WinogradFilterTransformOp : Linalg_Op<"winograd_filter_transform",
+    [DeclareOpInterfaceMethods<TilingInterface,
+     ["getIterationDomain",
+      "getLoopIteratorTypes",
+      "getResultTilePosition",
+      "getTiledImplementation"]>]> {
+  let summary = "Winograd filter transform operator";
+  let description = [{
+    linalg.winograd_filter_transform transforms the filter of conv2D.
----------------
ftynse wrote:

Please provide a more extended description, pseudo-code showing computation before/after or equivalent loop-level IR is welcome here.

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


More information about the Mlir-commits mailing list