File tree Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Original file line number Diff line number Diff line change @@ -22,8 +22,8 @@ class Device:
22
22
23
23
This is a wrapper around :class:`dpctl.SyclQueue` with custom
24
24
formatting. The class does not have public constructor,
25
- but a class method to construct it from device= keyword
26
- in Array-API functions.
25
+ but a class method `create_device` to construct it from device= keyword
26
+ argument in Array-API functions.
27
27
28
28
Instance can be queried for ``sycl_queue``, ``sycl_context``,
29
29
or ``sycl_device``.
@@ -111,6 +111,12 @@ def __repr__(self):
111
111
# This is a sub-device
112
112
return repr (self .sycl_queue )
113
113
114
+ def wait (self ):
115
+ """
116
+ Call ``wait`` method of the underlying ``sycl_queue``.
117
+ """
118
+ self .sycl_queue_ .wait ()
119
+
114
120
115
121
def normalize_queue_device (sycl_queue = None , device = None ):
116
122
"""
You can’t perform that action at this time.
0 commit comments