@@ -49,7 +49,7 @@ def _isdtype_impl(dtype, kind):
49
49
raise TypeError (f"Unsupported data type kind: { kind } " )
50
50
51
51
52
- __array_api_version__ = "2022 .12"
52
+ __array_api_version__ = "2023 .12"
53
53
54
54
55
55
class Info :
@@ -80,6 +80,8 @@ def __init__(self):
80
80
81
81
def capabilities (self ):
82
82
"""
83
+ capabilities()
84
+
83
85
Returns a dictionary of `dpctl`'s capabilities.
84
86
85
87
Returns:
@@ -92,12 +94,16 @@ def capabilities(self):
92
94
93
95
def default_device (self ):
94
96
"""
97
+ default_device()
98
+
95
99
Returns the default SYCL device.
96
100
"""
97
101
return dpctl .select_default_device ()
98
102
99
- def default_dtypes (self , device = None ):
103
+ def default_dtypes (self , * , device = None ):
100
104
"""
105
+ default_dtypes(*, device=None)
106
+
101
107
Returns a dictionary of default data types for `device`.
102
108
103
109
Args:
@@ -129,8 +135,10 @@ def default_dtypes(self, device=None):
129
135
"indexing" : dpt .dtype (default_device_index_type (device )),
130
136
}
131
137
132
- def dtypes (self , device = None , kind = None ):
138
+ def dtypes (self , * , device = None , kind = None ):
133
139
"""
140
+ dtypes(*, device=None, kind=None)
141
+
134
142
Returns a dictionary of all Array API data types of a specified `kind`
135
143
supported by `device`
136
144
@@ -193,13 +201,16 @@ def dtypes(self, device=None, kind=None):
193
201
194
202
def devices (self ):
195
203
"""
204
+ devices()
205
+
196
206
Returns a list of supported devices.
197
207
"""
198
208
return dpctl .get_devices ()
199
209
200
210
201
211
def __array_namespace_info__ ():
202
- """__array_namespace_info__()
212
+ """
213
+ __array_namespace_info__()
203
214
204
215
Returns a namespace with Array API namespace inspection utilities.
205
216
0 commit comments