We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 949711e commit b0fa998Copy full SHA for b0fa998
dpctl/tensor/_device.py
@@ -124,6 +124,18 @@ def wait(self):
124
"""
125
self.sycl_queue_.wait()
126
127
+ def __eq__(self, other):
128
+ """Equality comparison based on underlying ``sycl_queue``."""
129
+ if isinstance(other, Device):
130
+ return self.sycl_queue.__eq__(other.sycl_queue)
131
+ elif isinstance(other, dpctl.SyclQueue):
132
+ return self.sycl_queue.__eq__(other)
133
+ return False
134
+
135
+ def __hash__(self):
136
+ """Compute object's hash value."""
137
+ return self.sycl_queue.__hash__()
138
139
140
def normalize_queue_device(sycl_queue=None, device=None):
141
0 commit comments