Skip to content

Commit 47df265

Browse files
chsiggvwbaker
authored andcommitted
1 parent 4f10fc0 commit 47df265

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lib/Dialect/Triton/Transforms/Combine.td

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ def CombineDotAddIPattern : Pat<
1717
[(Constraint<CPred<"isZero($0)">> $c),
1818
(Constraint<CPred<"res->hasOneUse()">, "dot result has a single use">)]>;
1919
def CombineDotAddFPattern : Pat<
20-
(Arith_AddFOp $d, (TT_DotOp:$res $a, $b, $c, $inputPrecision, $maxNumImpreciseAcc), $fastmath),
20+
(Arith_AddFOp $d, (TT_DotOp:$res $a, $b, $c, $inputPrecision, $maxNumImpreciseAcc), $fastmath, $denorm),
2121
(TT_DotOp $a, $b, $d, $inputPrecision, $maxNumImpreciseAcc, (location $res)),
2222
[(Constraint<CPred<"isZero($0)">> $c),
2323
(Constraint<CPred<"::llvm::cast<::mlir::IntegerAttr>($0).getInt() == 0">> $maxNumImpreciseAcc),
@@ -29,7 +29,7 @@ def CombineDotAddIRevPattern : Pat<
2929
[(Constraint<CPred<"isZero($0)">> $c),
3030
(Constraint<CPred<"res->hasOneUse()">, "dot result has a single use">)]>;
3131
def CombineDotAddFRevPattern : Pat<
32-
(Arith_AddFOp (TT_DotOp:$res $a, $b, $c, $inputPrecision, $maxNumImpreciseAcc), $d, $fastmath),
32+
(Arith_AddFOp (TT_DotOp:$res $a, $b, $c, $inputPrecision, $maxNumImpreciseAcc), $d, $fastmath, $denorm),
3333
(TT_DotOp $a, $b, $d, $inputPrecision, $maxNumImpreciseAcc, (location $res)),
3434
[(Constraint<CPred<"isZero($0)">> $c),
3535
(Constraint<CPred<"::llvm::cast<::mlir::IntegerAttr>($0).getInt() == 0">> $maxNumImpreciseAcc),

0 commit comments

Comments
 (0)