Skip to content

Commit cf17eac

Browse files
committed
pass scalars to tensor.searchsorted directly
1 parent fc69816 commit cf17eac

1 file changed

Lines changed: 3 additions & 5 deletions

File tree

dpnp/dpnp_iface_searching.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -374,14 +374,12 @@ def searchsorted(a, v, side="left", sorter=None):
374374
"""
375375

376376
usm_a = dpnp.get_usm_ndarray(a)
377-
if dpnp.isscalar(v):
378-
usm_v = dpt.asarray(v, sycl_queue=a.sycl_queue, usm_type=a.usm_type)
379-
else:
380-
usm_v = dpnp.get_usm_ndarray(v)
377+
if not dpnp.isscalar(v):
378+
v = dpnp.get_usm_ndarray(v)
381379

382380
usm_sorter = None if sorter is None else dpnp.get_usm_ndarray(sorter)
383381
return dpnp_array._create_from_usm_ndarray(
384-
dpt.searchsorted(usm_a, usm_v, side=side, sorter=usm_sorter)
382+
dpt.searchsorted(usm_a, v, side=side, sorter=usm_sorter)
385383
)
386384

387385

0 commit comments

Comments
 (0)