[Relax][PyTorch] Add support for abs, ceil, erf, floor, log ops and refactor unary tests

python/tvm/relax/frontend/torch/fx_translator.py

@@ -616,21 +616,26 @@ def create_convert_map(
             nn.Flatten: self._flatten_module,
             ## call_function and call_method
             # unary
+            "abs": self._unary_op(relax.op.abs),
             "acos": self._unary_op(relax.op.acos),
             "acosh": self._unary_op(relax.op.acosh),
             "asin": self._unary_op(relax.op.asin),
             "asinh": self._unary_op(relax.op.asinh),
             "atan": self._unary_op(relax.op.atan),
             "atanh": self._unary_op(relax.op.atanh),
+            "ceil": self._unary_op(relax.op.ceil),
             "clamp": self._clamp,
             "cos": self._unary_op(relax.op.cos),
             "cosh": self._unary_op(relax.op.cosh),
             "dropout": lambda node: self.env[node.args[0]],
+            "erf": self._unary_op(relax.op.erf),
             "exp": self._unary_op(relax.op.exp),
+            "floor": self._unary_op(relax.op.floor),
             "gelu": self._gelu,
             "hardsigmoid": self._hardsigmoid,
             "hardswish": self._hardswish,
             "leaky_relu": self._leakyrelu,
+            "log": self._unary_op(relax.op.log),
             "log_softmax": self._log_softmax,
             "neg": self._unary_op(relax.op.negative),
             "relu": self._unary_op(relax.op.nn.relu),