Skip to content

Commit fd21828

Browse files
Resolves gh-1738
The unique_inverse and unique_all now always return inverse_index data fields in default indexing data type as per Python Array API specification.
1 parent d191e97 commit fd21828

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

dpctl/tensor/_set_functions.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -425,8 +425,7 @@ def unique_inverse(x):
425425
)
426426
_manager.add_event_pair(ht_ev, sub_ev)
427427

428-
inv_dt = dpt.int64 if x.size > dpt.iinfo(dpt.int32).max else dpt.int32
429-
inv = dpt.empty_like(x, dtype=inv_dt, order="C")
428+
inv = dpt.empty_like(x, dtype=ind_dt, order="C")
430429
ht_ev, ssl_ev = _searchsorted_left(
431430
hay=unique_vals,
432431
needles=x,
@@ -608,8 +607,7 @@ def unique_all(x: dpt.usm_ndarray) -> UniqueAllResult:
608607
)
609608
_manager.add_event_pair(ht_ev, sub_ev)
610609

611-
inv_dt = dpt.int64 if x.size > dpt.iinfo(dpt.int32).max else dpt.int32
612-
inv = dpt.empty_like(x, dtype=inv_dt, order="C")
610+
inv = dpt.empty_like(x, dtype=ind_dt, order="C")
613611
ht_ev, ssl_ev = _searchsorted_left(
614612
hay=unique_vals,
615613
needles=x,

0 commit comments

Comments
 (0)