Skip to content

Commit 85ccecb

Browse files
committed
Add missing host task to host_tasks_list in _reduction.py
1 parent e45c68e commit 85ccecb

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

dpctl/tensor/_reduction.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -164,6 +164,7 @@ def _reduction_over_axis(
164164
sycl_queue=q,
165165
depends=[cpy_e],
166166
)
167+
host_tasks_list.append(ht_e_red)
167168
ht_e_cpy2, _ = ti._copy_usm_ndarray_into_usm_ndarray(
168169
src=tmp_res, dst=res, sycl_queue=q, depends=[r_e]
169170
)

0 commit comments

Comments
 (0)