Skip to content

Commit cc6b5c3

Browse files
committed
remove unnecessary copy
1 parent cf17eac commit cc6b5c3

1 file changed

Lines changed: 3 additions & 4 deletions

File tree

dpnp/tensor/_searchsorted.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -171,9 +171,6 @@ def searchsorted(
171171
dt1, dt2 = _resolve_weak_types_all_py_ints(x1_dt, x2_dt, sycl_dev)
172172
dt = _to_device_supported_dtype(dpt.result_type(dt1, dt2), sycl_dev)
173173

174-
if not isinstance(x2, usm_ndarray):
175-
x2 = dpt.asarray(x2, dtype=dt2, usm_type=res_usm_type, sycl_queue=q)
176-
177174
# get submitted events again in case some were added by sorter handling
178175
dep_evs = _manager.submitted_events
179176
if x1_dt != dt:
@@ -182,7 +179,9 @@ def searchsorted(
182179
_manager.add_event_pair(ht_ev, ev)
183180
x1 = x1_buf
184181

185-
if x2.dtype != dt:
182+
if not isinstance(x2, usm_ndarray):
183+
x2 = dpt.asarray(x2, dtype=dt2, usm_type=res_usm_type, sycl_queue=q)
184+
elif x2_dt != dt:
186185
x2_buf = _empty_like_orderK(x2, dt)
187186
ht_ev, ev = ti_copy(src=x2, dst=x2_buf, sycl_queue=q, depends=dep_evs)
188187
_manager.add_event_pair(ht_ev, ev)

0 commit comments

Comments
 (0)