[Mlir-commits] [mlir] [mlir][linalg] Fix for bias handling for Winograd (PR #110331)
Dmitriy Smirnov
llvmlistbot at llvm.org
Sat Sep 28 01:03:39 PDT 2024
================
@@ -837,9 +837,25 @@ Value outputTransform(RewriterBase &rewriter, Location loc, Value value,
Value widthOffset =
builder.create<affine::AffineApplyOp>(loc, affineMap, tileWIter);
+ // Handling bias.
----------------
d-smirnov wrote:
The output tensor is padded, pre-initialized with bias values, and supplied to output transform op as an output parameter. However the pre-initialized values will be discarded later when the output transformation actually happens. The patch fixes the issue.
https://github.com/llvm/llvm-project/pull/110331
More information about the Mlir-commits
mailing list