Skip to content

Commit 56bb65f

Browse files
Moved ctx creation into host-task-dispatching handler function.
1 parent 84ba81a commit 56bb65f

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

dpctl/tensor/libtensor/source/integer_advanced_indexing.cpp

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -677,9 +677,9 @@ usm_ndarray_take(dpctl::tensor::usm_ndarray src,
677677
src_offset, dst_offset, packed_ind_offsets, all_deps);
678678

679679
// free packed temporaries
680-
auto ctx = exec_q.get_context();
681680
exec_q.submit([&](sycl::handler &cgh) {
682681
cgh.depends_on(take_generic_ev);
682+
auto ctx = exec_q.get_context();
683683
cgh.host_task([packed_shapes_strides, packed_axes_shapes_strides,
684684
packed_ind_shapes_strides, packed_ind_ptrs,
685685
packed_ind_offsets, ctx]() {
@@ -691,9 +691,10 @@ usm_ndarray_take(dpctl::tensor::usm_ndarray src,
691691
});
692692
});
693693

694-
return std::make_pair(
695-
keep_args_alive(exec_q, {src, py_ind, dst}, {take_generic_ev}),
696-
take_generic_ev);
694+
sycl::event host_task_ev =
695+
keep_args_alive(exec_q, {src, py_ind, dst}, {take_generic_ev});
696+
697+
return std::make_pair(host_task_ev, take_generic_ev);
697698
}
698699

699700
std::pair<sycl::event, sycl::event>

0 commit comments

Comments
 (0)