Finished trigonometric functions (#21128)

Chillee · facebook-github-bot · commit 92b76df8f699 · 2019-06-04T17:59:09.000-07:00
Summary: Pull Request resolved: #21128 ghimport-source-id: d566de1 Differential Revision: D15563190 Pulled By: Chillee fbshipit-source-id: ad2e09cac5c7dae9978a7bd61098c2828620cdc4
diff --git a/test/test_jit.py b/test/test_jit.py
@@ -6127,9 +6127,9 @@ def func(a, b):
                            .format(func_name=func_name, a=a, b=b, res_python=res_python, res_script=res_script))
                     self.assertEqual(res_python, res_script, message=msg, prec=(1e-4) * max(abs(res_python), res_script))
 
-
         unary_float_ops = ["log", "log1p", "log10", "exp", "sqrt", "gamma", "lgamma", "erf",
-                           "erfc", "expm1", "fabs", "acos", "asin", "atan", "cos", "sin", "tan"]
+                           "erfc", "expm1", "fabs", "acos", "asin", "atan", "cos", "sin", "tan",
+                           "asinh", "atanh", "acosh", "sinh", "cosh", "tanh"]
         binary_float_ops = ["atan2", "fmod", "copysign"]
         for op in unary_float_ops:
             checkMathWrap(op)
diff --git a/torch/csrc/jit/register_prim_ops.cpp b/torch/csrc/jit/register_prim_ops.cpp
@@ -2160,6 +2160,12 @@ RegisterOperators reg2({
     DEFINE_UNARY_OP(aten::cos, std::cos(a), float, float),
     DEFINE_UNARY_OP(aten::sin, std::sin(a), float, float),
     DEFINE_UNARY_OP(aten::tan, std::tan(a), float, float),
+    DEFINE_UNARY_OP(aten::asinh, std::asinh(a), float, float),
+    DEFINE_UNARY_OP(aten::atanh, std::atanh(a), float, float),
+    DEFINE_UNARY_OP(aten::acosh, std::acosh(a), float, float),
+    DEFINE_UNARY_OP(aten::sinh, std::sinh(a), float, float),
+    DEFINE_UNARY_OP(aten::cosh, std::cosh(a), float, float),
+    DEFINE_UNARY_OP(aten::tanh, std::tanh(a), float, float),
     DEFINE_BINARY_FLOAT_OP(aten::fmod, std::fmod(a, b)),
     Operator(
         "aten::modf(float a) -> (float, float)",
diff --git a/torch/jit/__init__.py b/torch/jit/__init__.py
@@ -1900,6 +1900,12 @@ def register_all(mod):
         (math.cos, "aten::cos"),
         (math.sin, "aten::sin"),
         (math.tan, "aten::tan"),
+        (math.asinh, "aten::asinh"),
+        (math.atanh, "aten::atanh"),
+        (math.acosh, "aten::acosh"),
+        (math.sinh, "aten::sinh"),
+        (math.cosh, "aten::cosh"),
+        (math.tanh, "aten::tanh"),
         (math.fmod, "aten::fmod"),
         (math.modf, "aten::modf"),
         (torch._C._infer_size, "aten::_infer_size"),
