Skip to content

Commit

Permalink
Fix the tt to linalg error in matmul converter,
Browse files Browse the repository at this point in the history
the tt.dot with accmulator will lower to linalg.matmul and arith.add,
and the arith.add will further lower to linalg.generic,
generic will take the lhs of add as the DPS init, so the lhs of add
must be the accmulator.
  • Loading branch information
MercuryChen committed Nov 21, 2024
1 parent 3fe82cb commit 713f325
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -1027,9 +1027,9 @@ struct MatmulConverter : public OpConversionPattern<triton::DotOp> {

if (!skipC) {
if (integers) {
res = rewriter.create<arith::AddIOp>(loc, res, opc);
res = rewriter.create<arith::AddIOp>(loc, opc, res);
} else {
res = rewriter.create<arith::AddFOp>(loc, res, opc);
res = rewriter.create<arith::AddFOp>(loc, opc, res);
}
}

Expand Down

0 comments on commit 713f325

Please sign in to comment.