Make the basic renames simple aliases instead of wrappers

This removes the overhead for these functions, and also makes it so that they
are still NumPy ufuncs with all the corresponding ufunc keywords and methods.

Author: asmeurer

array_api_compat/common/_aliases.py

@@ -14,43 +14,6 @@
 
 from ._helpers import _check_device, _is_numpy_array, get_namespace
 
-# Basic renames
-def acos(x, /, xp):
-    return xp.arccos(x)
-
-def acosh(x, /, xp):
-    return xp.arccosh(x)
-
-def asin(x, /, xp):
-    return xp.arcsin(x)
-
-def asinh(x, /, xp):
-    return xp.arcsinh(x)
-
-def atan(x, /, xp):
-    return xp.arctan(x)
-
-def atan2(x1, x2, /, xp):
-    return xp.arctan2(x1, x2)
-
-def atanh(x, /, xp):
-    return xp.arctanh(x)
-
-def bitwise_left_shift(x1, x2, /, xp):
-    return xp.left_shift(x1, x2)
-
-def bitwise_invert(x, /, xp):
-    return xp.invert(x)
-
-def bitwise_right_shift(x1, x2, /, xp):
-    return xp.right_shift(x1, x2)
-
-def concat(arrays: Union[Tuple[ndarray, ...], List[ndarray]], /, xp, *, axis: Optional[int] = 0) -> ndarray:
-    return xp.concatenate(arrays, axis=axis)
-
-def pow(x1, x2, /, xp):
-    return xp.power(x1, x2)
-
 # These functions are modified from the NumPy versions.
 
 def arange(
@@ -422,10 +385,7 @@ def trunc(x: ndarray, /, xp) -> ndarray:
         return x
     return xp.trunc(x)
 
-__all__ = ['acos', 'acosh', 'asin', 'asinh', 'atan', 'atan2', 'atanh',
-           'bitwise_left_shift', 'bitwise_invert', 'bitwise_right_shift',
-           'concat', 'pow', 'UniqueAllResult', 'UniqueCountsResult',
-           'UniqueInverseResult', 'unique_all', 'unique_counts',
-           'unique_inverse', 'unique_values', 'astype', 'std', 'var',
-           'permute_dims', 'reshape', 'argsort', 'sort', 'sum', 'prod',
-           'ceil', 'floor', 'trunc']
+__all__ = ['UniqueAllResult', 'UniqueCountsResult', 'UniqueInverseResult',
+           'unique_all', 'unique_counts', 'unique_inverse', 'unique_values',
+           'astype', 'std', 'var', 'permute_dims', 'reshape', 'argsort',
+           'sort', 'sum', 'prod', 'ceil', 'floor', 'trunc']

array_api_compat/cupy/_aliases.py

@@ -13,18 +13,20 @@
 import cupy as cp
 bool = cp.bool_
 
-acos = get_xp(cp)(_aliases.acos)
-acosh = get_xp(cp)(_aliases.acosh)
-asin = get_xp(cp)(_aliases.asin)
-asinh = get_xp(cp)(_aliases.asinh)
-atan = get_xp(cp)(_aliases.atan)
-atan2 = get_xp(cp)(_aliases.atan2)
-atanh = get_xp(cp)(_aliases.atanh)
-bitwise_left_shift = get_xp(cp)(_aliases.bitwise_left_shift)
-bitwise_invert = get_xp(cp)(_aliases.bitwise_invert)
-bitwise_right_shift = get_xp(cp)(_aliases.bitwise_right_shift)
-concat = get_xp(cp)(_aliases.concat)
-pow = get_xp(cp)(_aliases.pow)
+# Basic renames
+acos = cp.arccos
+acosh = cp.arccosh
+asin = cp.arcsin
+asinh = cp.arcsinh
+atan = cp.arctan
+atan2 = cp.arctan2
+atanh = cp.arctanh
+bitwise_left_shift = cp.left_shift
+bitwise_invert = cp.invert
+bitwise_right_shift = cp.right_shift
+concat = cp.concatenate
+pow = cp.power
+
 arange = get_xp(cp)(_aliases.arange)
 empty = get_xp(cp)(_aliases.empty)
 empty_like = get_xp(cp)(_aliases.empty_like)

array_api_compat/numpy/_aliases.py

@@ -13,18 +13,20 @@
 import numpy as np
 bool = np.bool_
 
-acos = get_xp(np)(_aliases.acos)
-acosh = get_xp(np)(_aliases.acosh)
-asin = get_xp(np)(_aliases.asin)
-asinh = get_xp(np)(_aliases.asinh)
-atan = get_xp(np)(_aliases.atan)
-atan2 = get_xp(np)(_aliases.atan2)
-atanh = get_xp(np)(_aliases.atanh)
-bitwise_left_shift = get_xp(np)(_aliases.bitwise_left_shift)
-bitwise_invert = get_xp(np)(_aliases.bitwise_invert)
-bitwise_right_shift = get_xp(np)(_aliases.bitwise_right_shift)
-concat = get_xp(np)(_aliases.concat)
-pow = get_xp(np)(_aliases.pow)
+# Basic renames
+acos = np.arccos
+acosh = np.arccosh
+asin = np.arcsin
+asinh = np.arcsinh
+atan = np.arctan
+atan2 = np.arctan2
+atanh = np.arctanh
+bitwise_left_shift = np.left_shift
+bitwise_invert = np.invert
+bitwise_right_shift = np.right_shift
+concat = np.concatenate
+pow = np.power
+
 arange = get_xp(np)(_aliases.arange)
 empty = get_xp(np)(_aliases.empty)
 empty_like = get_xp(np)(_aliases.empty_like)