[Mlir-commits] [mlir] [mlir][linalg] Implement TilingInterface for winograd operators (PR #96184)

llvmlistbot at llvm.org llvmlistbot at llvm.org
Fri Jul 19 08:02:21 PDT 2024


================
@@ -154,8 +154,8 @@ def Linalg_SoftmaxOp : Linalg_Op<"softmax",
   let hasVerifier = 1;
 }
 
-def Linalg_WinogradFilterTransformOp :
-    Linalg_Op<"winograd_filter_transform", [AllElementTypesMatch<["filter", "output"]>]> {
+def Linalg_WinogradFilterTransformOp : Linalg_Op<"winograd_filter_transform",
+    [AllElementTypesMatch<["filter", "output"]>]> {
----------------
Max191 wrote:

Can you implement the tiling interface for the filter transform as well? I think it would look similar to the input transform tiled implementation, but instead of extracting a tile of the input, it extracts the full filter size.

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


More information about the Mlir-commits mailing list