Skip to content

Commit 11cef65

Browse files
authored
One more erf op (#2223)
1 parent e359b8a commit 11cef65

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/enzyme_ad/jax/Passes/AffineToStableHLORaising.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2555,7 +2555,8 @@ tryRaisingOpToStableHLO(Operation *op, IRMapping &mapping, OpBuilder &builder,
25552555
arith::ExtUIOp, arith::SIToFPOp, arith::UIToFPOp, arith::FPToSIOp,
25562556
arith::FPToUIOp, arith::TruncFOp, arith::ExtFOp, math::SqrtOp,
25572557
math::RsqrtOp, math::CbrtOp, math::LogOp, math::ExpOp, math::AbsFOp,
2558-
math::AbsIOp, math::IsNaNOp, math::AtanOp, arith::BitcastOp>(op)) {
2558+
math::AbsIOp, math::IsNaNOp, math::AtanOp, arith::BitcastOp,
2559+
math::ErfOp>(op)) {
25592560
assert(op->getNumOperands() == 1 && op->getNumResults() == 1);
25602561

25612562
auto operand = op->getOperand(0);

0 commit comments

Comments
 (0)