Skip to content

Commit 6a3d9e8

Browse files
Order copy into order='K' allocation task after tasks populating source array
1 parent fbb162c commit 6a3d9e8

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

dpctl/tensor/_ctors.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1472,9 +1472,10 @@ def full_like(
14721472
X = dpt.broadcast_to(X, sh)
14731473
res = _empty_like_orderK(x, dtype, usm_type, sycl_queue)
14741474
_manager = dpctl.utils.SequentialOrderManager[sycl_queue]
1475-
# populating new allocation, no dependent events
1475+
# order copy after tasks populating X
1476+
dep_evs = _manager.submitted_events
14761477
hev, copy_ev = ti._copy_usm_ndarray_into_usm_ndarray(
1477-
src=X, dst=res, sycl_queue=sycl_queue
1478+
src=X, dst=res, sycl_queue=sycl_queue, depends=dep_evs
14781479
)
14791480
_manager.add_event_pair(hev, copy_ev)
14801481
return res

0 commit comments

Comments
 (0)