support erf/asinh/acosh/atanh

Signed-off-by: inocsin <[email protected]>

Author: inocsin

core/conversion/converters/impl/unary.cpp

@@ -40,6 +40,10 @@ convert(ceil, kCEIL);
 convert(sqrt, kSQRT);
 convert(exp, kEXP);
 convert(neg, kNEG);
+convert(erf, kERF);
+convert(asinh, kASINH);
+convert(acosh, kACOSH);
+convert(atanh, kATANH);
 
 #undef convert
 

tests/core/conversion/converters/test_unary.cpp

@@ -21,7 +21,9 @@ std::string gen_test_graph(const std::string& unary) {
     auto g = std::make_shared<torch::jit::Graph>();                                       \
     torch::jit::parseIR(graph, &*g);                                                      \
                                                                                           \
-    auto in = at::empty({10}, {at::kCUDA}).uniform_(0, 0.5);                              \
+    float offset = 0;                                                                     \
+    if (#name == "Acosh") offset += 1;   /*input larger than 1 for acosh*/                \
+    auto in = at::empty({10}, {at::kCUDA}).uniform_(0+offset, 0.5+offset);                \
     auto params = trtorch::core::conversion::get_named_params(g->inputs(), {});           \
     auto jit_results = trtorch::tests::util::RunGraph(g, params, {in});                   \
                                                                                           \
@@ -48,5 +50,9 @@ test_unary(ceil, Ceil);
 test_unary(sqrt, Sqrt);
 test_unary(exp, Exp);
 test_unary(neg, Neg);
+test_unary(erf, Erf);
+test_unary(asinh, Asinh);
+test_unary(acosh, Acosh);
+test_unary(atanh, Atanh);
 
 #undef test_unary