Skip to content

Commit

Permalink
Remove wrong forward diff rule of chlo.acos
Browse files Browse the repository at this point in the history
  • Loading branch information
mofeing committed Jun 20, 2024
1 parent 1443121 commit 9c5c0e6
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/enzyme_ad/jax/Implementations/CHLODerivatives.td
Original file line number Diff line number Diff line change
Expand Up @@ -93,7 +93,7 @@ def TopK : HLOInst<"TopKOp">;
// Derivative rules
def : HLODerivative<"AcosOp", (Op $x), [
(Neg (Div (DiffeRet), (Sqrt (Sub (HLOConstantFP<"1"> $x), (Pow $x, (HLOConstantFP<"2"> $x))))))
], (Acos (Shadow $x))>;
]>;

def : HLODerivative<"AcoshOp", (Op $x), [
(Div (DiffeRet), (Mul (Sqrt (Sub (Pow $x, (HLOConstantFP<"2"> $x)), (HLOConstantFP<"1"> $x))), (Sqrt (Add (Pow $x, (HLOConstantFP<"2"> $x)), (HLOConstantFP<"1"> $x)))))
Expand Down

0 comments on commit 9c5c0e6

Please sign in to comment.