Update the function stubs with device support

Author: asmeurer

array_api_tests/function_stubs/__init__.py

@@ -9,9 +9,9 @@
 
 __all__ = []
 
-from .array_object import __abs__, __add__, __and__, __eq__, __floordiv__, __ge__, __getitem__, __gt__, __invert__, __le__, __len__, __lshift__, __lt__, __matmul__, __mod__, __mul__, __ne__, __neg__, __or__, __pos__, __pow__, __rshift__, __setitem__, __sub__, __truediv__, __xor__, dtype, ndim, shape, size, T
+from .array_object import __abs__, __add__, __and__, __eq__, __floordiv__, __ge__, __getitem__, __gt__, __invert__, __le__, __len__, __lshift__, __lt__, __matmul__, __mod__, __mul__, __ne__, __neg__, __or__, __pos__, __pow__, __rshift__, __setitem__, __sub__, __truediv__, __xor__, dtype, device, ndim, shape, size, T
 
-__all__ += ['__abs__', '__add__', '__and__', '__eq__', '__floordiv__', '__ge__', '__getitem__', '__gt__', '__invert__', '__le__', '__len__', '__lshift__', '__lt__', '__matmul__', '__mod__', '__mul__', '__ne__', '__neg__', '__or__', '__pos__', '__pow__', '__rshift__', '__setitem__', '__sub__', '__truediv__', '__xor__', 'dtype', 'ndim', 'shape', 'size', 'T']
+__all__ += ['__abs__', '__add__', '__and__', '__eq__', '__floordiv__', '__ge__', '__getitem__', '__gt__', '__invert__', '__le__', '__len__', '__lshift__', '__lt__', '__matmul__', '__mod__', '__mul__', '__ne__', '__neg__', '__or__', '__pos__', '__pow__', '__rshift__', '__setitem__', '__sub__', '__truediv__', '__xor__', 'dtype', 'device', 'ndim', 'shape', 'size', 'T']
 
 from .constants import e, inf, nan, pi
 

array_api_tests/function_stubs/array_object.py

@@ -173,6 +173,9 @@ def __xor__(x1, x2):
 # Note: dtype is an attribute of the array object.
 dtype = None
 
+# Note: device is an attribute of the array object.
+device = None
+
 # Note: ndim is an attribute of the array object.
 ndim = None
 
@@ -185,4 +188,4 @@ def __xor__(x1, x2):
 # Note: T is an attribute of the array object.
 T = None
 
-__all__ = ['__abs__', '__add__', '__and__', '__eq__', '__floordiv__', '__ge__', '__getitem__', '__gt__', '__invert__', '__le__', '__len__', '__lshift__', '__lt__', '__matmul__', '__mod__', '__mul__', '__ne__', '__neg__', '__or__', '__pos__', '__pow__', '__rshift__', '__setitem__', '__sub__', '__truediv__', '__xor__', 'dtype', 'ndim', 'shape', 'size', 'T']
+__all__ = ['__abs__', '__add__', '__and__', '__eq__', '__floordiv__', '__ge__', '__getitem__', '__gt__', '__invert__', '__le__', '__len__', '__lshift__', '__lt__', '__matmul__', '__mod__', '__mul__', '__ne__', '__neg__', '__or__', '__pos__', '__pow__', '__rshift__', '__setitem__', '__sub__', '__truediv__', '__xor__', 'dtype', 'device', 'ndim', 'shape', 'size', 'T']

array_api_tests/function_stubs/creation_functions.py

@@ -14,37 +14,37 @@
 2. There is no real way to test that anyway.
 """
 
-def arange(start, *, stop=None, step=1, dtype=None):
+def arange(start, *, stop=None, step=1, dtype=None, device=None):
     pass
 
-def empty(shape, *, dtype=None):
+def empty(shape, *, dtype=None, device=None):
     pass
 
-def empty_like(x, *, dtype=None):
+def empty_like(x, *, dtype=None, device=None):
     pass
 
-def eye(N, *, M=None, k=0, dtype=None):
+def eye(N, *, M=None, k=0, dtype=None, device=None):
     pass
 
-def full(shape, fill_value, *, dtype=None):
+def full(shape, fill_value, *, dtype=None, device=None):
     pass
 
-def full_like(x, fill_value, *, dtype=None):
+def full_like(x, fill_value, *, dtype=None, device=None):
     pass
 
-def linspace(start, stop, num, *, dtype=None, endpoint=True):
+def linspace(start, stop, num, *, dtype=None, device=None, endpoint=True):
     pass
 
-def ones(shape, *, dtype=None):
+def ones(shape, *, dtype=None, device=None):
     pass
 
-def ones_like(x, *, dtype=None):
+def ones_like(x, *, dtype=None, device=None):
     pass
 
-def zeros(shape, *, dtype=None):
+def zeros(shape, *, dtype=None, device=None):
     pass
 
-def zeros_like(x, *, dtype=None):
+def zeros_like(x, *, dtype=None, device=None):
     pass
 
 __all__ = ['arange', 'empty', 'empty_like', 'eye', 'full', 'full_like', 'linspace', 'ones', 'ones_like', 'zeros', 'zeros_like']