Add __nv_fast_expf to InstructionDerivatives.td (#2068)

* Add __nv_fast_expf to InstructionDerivatives.td * Update InstructionDerivatives.td
EnzymeAD · Sep 3, 2024 · ce81c9c · ce81c9c
1 parent 457320d
commit ce81c9c
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/enzyme/Enzyme/InstructionDerivatives.td b/enzyme/Enzyme/InstructionDerivatives.td
@@ -413,6 +413,13 @@ def : CallPattern<(Op $x),
                   [ReadNone, NoUnwind]
                   >;
 
+def : CallPattern<(Op $x),
+                  ["__nv_fast_expf"],
+                  [(FMul (DiffeRet), (Call<(SameFunc), [ReadNone,NoUnwind]> $x) )],
+                  (ForwardFromSummedReverse),
+                  [ReadNone, NoUnwind]
+                  >;
+
 def : CallPattern<(Op $x),
                   ["exp10"],
                   [(FMul (FMul (DiffeRet), (Call<(SameFunc), [ReadNone,NoUnwind]> $x) ), (ConstantFP<"2.30258509299404568401799145468"> $x))],