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

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


================
@@ -2760,6 +2760,89 @@ LogicalResult WinogradFilterTransformOp::verify() {
   return success();
 }
 
+SmallVector<Range>
+WinogradFilterTransformOp::getIterationDomain(OpBuilder &builder) {
+  Location loc = getLoc();
+  Value zero = builder.create<arith::ConstantIndexOp>(loc, 0);
+  Value one = builder.create<arith::ConstantIndexOp>(loc, 1);
----------------
ftynse wrote:

IIRC, `Range` contains list of `OpFoldResult`, meaning we can put attributes there and not materialize operations for these constants.

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


More information about the llvm-branch-commits mailing list