Move ti.acos()/acosh() and reuse them in dpnp

Author: vlad-perevezentsev

dpctl_ext/tensor/__init__.py

@@ -85,6 +85,8 @@
 from ._clip import clip
 from ._elementwise_funcs import (
     abs,
+    acos,
+    acosh,
 )
 from ._reduction import (
     argmax,
@@ -110,6 +112,8 @@
 
 __all__ = [
     "abs",
+    "acos",
+    "acosh",
     "all",
     "any",
     "arange",

dpctl_ext/tensor/_elementwise_funcs.py

@@ -61,3 +61,61 @@
 
 abs = UnaryElementwiseFunc("abs", ti._abs_result_type, ti._abs, _abs_docstring_)
 del _abs_docstring_
+
+# U02: ==== ACOS   (x)
+_acos_docstring = r"""
+acos(x, /, \*, out=None, order='K')
+
+Computes inverse cosine for each element `x_i` for input array `x`.
+
+Args:
+    x (usm_ndarray):
+        Input array, expected to have a floating-point data type.
+    out (Union[usm_ndarray, None], optional):
+        Output array to populate.
+        Array must have the correct shape and the expected data type.
+    order ("C","F","A","K", optional):
+        Memory layout of the new output array, if parameter
+        `out` is ``None``.
+        Default: "K".
+
+Returns:
+    usm_ndarray:
+        An array containing the element-wise inverse cosine, in radians
+        and in the closed interval :math:`[0, \pi]`. The data type of the
+        returned array is determined by the Type Promotion Rules.
+"""
+
+acos = UnaryElementwiseFunc(
+    "acos", ti._acos_result_type, ti._acos, _acos_docstring
+)
+del _acos_docstring
+
+# U03: ===== ACOSH (x)
+_acosh_docstring = r"""
+acosh(x, /, \*, out=None, order='K')
+
+Computes inverse hyperbolic cosine for each element `x_i` for input array `x`.
+
+Args:
+    x (usm_ndarray):
+        Input array, expected to have a floating-point data type.
+    out (Union[usm_ndarray, None], optional):
+        Output array to populate.
+        Array must have the correct shape and the expected data type.
+    order ("C","F","A","K", optional):
+        Memory layout of the new output array, if parameter
+        `out` is ``None``.
+        Default: "K".
+
+Returns:
+    usm_ndarray:
+        An array containing the element-wise inverse hyperbolic cosine, in
+        radians and in the half-closed interval :math:`[0, \infty)`. The data
+        type of the returned array is determined by the Type Promotion Rules.
+"""
+
+acosh = UnaryElementwiseFunc(
+    "acosh", ti._acosh_result_type, ti._acosh, _acosh_docstring
+)
+del _acosh_docstring

dpnp/dpnp_iface_trigonometric.py

@@ -47,6 +47,7 @@
 # TODO: revert to `import dpctl.tensor...`
 # when dpnp fully migrates dpctl/tensor
 import dpctl_ext.tensor as dpt
+import dpctl_ext.tensor._tensor_elementwise_impl as ti_ext
 import dpctl_ext.tensor._type_utils as dtu
 import dpnp
 import dpnp.backend.extensions.ufunc._ufunc_impl as ufi
@@ -138,8 +139,8 @@ def _get_accumulation_res_dt(a, dtype):
 
 acos = DPNPUnaryFunc(
     "acos",
-    ti._acos_result_type,
-    ti._acos,
+    ti_ext._acos_result_type,
+    ti_ext._acos,
     _ACOS_DOCSTRING,
     mkl_fn_to_call="_mkl_acos_to_call",
     mkl_impl_fn="_acos",
@@ -224,8 +225,8 @@ def _get_accumulation_res_dt(a, dtype):
 
 acosh = DPNPUnaryFunc(
     "acosh",
-    ti._acosh_result_type,
-    ti._acosh,
+    ti_ext._acosh_result_type,
+    ti_ext._acosh,
     _ACOSH_DOCSTRING,
     mkl_fn_to_call="_mkl_acosh_to_call",
     mkl_impl_fn="_acosh",