File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff 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 ,
You can’t perform that action at this time.
0 commit comments