Skip to content

Commit 1b9c2ec

Browse files
committed
do not sequence copy and cast for x1 and x2 in searchsorted
1 parent b7c56e1 commit 1b9c2ec

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dpctl/tensor/_searchsorted.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -139,9 +139,10 @@ def searchsorted(
139139
dt1, dt2 = _resolve_weak_types_all_py_ints(x1_dt, x2_dt, sycl_dev)
140140
dt = _to_device_supported_dtype(dpt.result_type(dt1, dt2), sycl_dev)
141141

142+
# get submitted events again in case some were added by sorter handling
143+
dep_evs = _manager.submitted_events
142144
if x1_dt != dt:
143145
x1_buf = _empty_like_orderK(x1, dt)
144-
dep_evs = _manager.submitted_events
145146
ht_ev, ev = ti_copy(src=x1, dst=x1_buf, sycl_queue=q, depends=dep_evs)
146147
_manager.add_event_pair(ht_ev, ev)
147148
x1 = x1_buf
@@ -150,7 +151,6 @@ def searchsorted(
150151
x2 = dpt.asarray(x2, dtype=dt2, usm_type=res_usm_type, sycl_queue=q)
151152
if x2.dtype != dt:
152153
x2_buf = _empty_like_orderK(x2, dt)
153-
dep_evs = _manager.submitted_events
154154
ht_ev, ev = ti_copy(src=x2, dst=x2_buf, sycl_queue=q, depends=dep_evs)
155155
_manager.add_event_pair(ht_ev, ev)
156156
x2 = x2_buf

0 commit comments

Comments
 (0)