Skip to content

Commit 6bc4ba8

Browse files
Added wait to avoid dangling host_task
1 parent c6ef075 commit 6bc4ba8

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

dpctl/tensor/_elementwise_common.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,7 @@ def __call__(self, x, order="K"):
8686
r = dpt.empty_like(buf, dtype=res_dt, order=order)
8787

8888
ht, _ = self.unary_fn_(buf, r, sycl_queue=exec_q, depends=[copy_ev])
89+
ht_copy_ev.wait()
8990
ht.wait()
9091

9192
return r

0 commit comments

Comments
 (0)