Skip to content

Commit d1a90bc

Browse files
authored
Unwrap dpnp array while calling dpctl indexer operators (#1321)
1 parent 44e9fa9 commit d1a90bc

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

dpnp/dpnp_array.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -174,6 +174,9 @@ def __ge__(self, other):
174174
# '__getattribute__',
175175

176176
def __getitem__(self, key):
177+
if isinstance(key, dpnp_array):
178+
key = key.get_array()
179+
177180
item = self._array_obj.__getitem__(key)
178181
if not isinstance(item, dpt.usm_ndarray):
179182
raise RuntimeError(
@@ -290,6 +293,11 @@ def __rtruediv__(self, other):
290293
# '__setattr__',
291294

292295
def __setitem__(self, key, val):
296+
if isinstance(key, dpnp_array):
297+
key = key.get_array()
298+
if isinstance(val, dpnp_array):
299+
val = val.get_array()
300+
293301
self._array_obj.__setitem__(key, val)
294302

295303
# '__setstate__',

0 commit comments

Comments
 (0)