Skip to content

Commit e5480b4

Browse files
authored
Add erf op in stablehlo raise (#2219)
1 parent 0a053b5 commit e5480b4

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/enzyme_ad/jax/Passes/ArithRaising.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,6 +122,7 @@ struct ArithRaisingPass
122122
RAISE_UNARY(math::IsFiniteOp, stablehlo::IsFiniteOp, mhlo::IsFiniteOp);
123123
RAISE_UNARY(math::CeilOp, stablehlo::CeilOp, mhlo::CeilOp);
124124
RAISE_UNARY(math::FloorOp, stablehlo::FloorOp, mhlo::FloorOp);
125+
RAISE_UNARY(math::ErfOp, chlo::ErfOp, chlo::ErfOp);
125126
RAISE_UNARY(arith::NegFOp, stablehlo::NegOp, mhlo::NegOp);
126127

127128
#undef RAISE_UNARY

0 commit comments

Comments
 (0)