@@ -229,6 +229,7 @@ std::vector<sycl::event> _populate_packed_shapes_strides_for_indexing(
229
229
cgh.host_task ([packed_host_axes_shapes_strides_shp,
230
230
packed_host_shapes_strides_shp]() {});
231
231
});
232
+ clean_up_host_task_ev.wait ();
232
233
host_task_events.push_back (clean_up_host_task_ev);
233
234
234
235
std::vector<sycl::event> v = {device_orthog_shapes_strides_copy_ev,
@@ -282,6 +283,7 @@ std::vector<sycl::event> _populate_packed_shapes_strides_for_indexing(
282
283
cgh.depends_on (device_axes_shapes_strides_copy_ev);
283
284
cgh.host_task ([packed_host_axes_shapes_strides_shp]() {});
284
285
});
286
+ clean_up_host_task_ev.wait ();
285
287
host_task_events.push_back (clean_up_host_task_ev);
286
288
287
289
std::vector<sycl::event> v = {device_orthog_shapes_strides_fill_ev,
@@ -627,6 +629,7 @@ usm_ndarray_take(dpctl::tensor::usm_ndarray src,
627
629
cgh.host_task ([host_ind_offsets_shp, host_ind_shapes_strides_shp,
628
630
host_ind_ptrs_shp]() {});
629
631
});
632
+ shared_ptr_cleanup_host_task.wait ();
630
633
host_task_events.push_back (shared_ptr_cleanup_host_task);
631
634
632
635
std::vector<sycl::event> ind_pack_depends{packed_ind_ptrs_copy_ev,
@@ -1039,6 +1042,7 @@ usm_ndarray_put(dpctl::tensor::usm_ndarray dst,
1039
1042
cgh.host_task ([host_ind_offsets_shp, host_ind_shapes_strides_shp,
1040
1043
host_ind_ptrs_shp]() {});
1041
1044
});
1045
+ shared_ptr_cleanup_host_task.wait ();
1042
1046
host_task_events.push_back (shared_ptr_cleanup_host_task);
1043
1047
1044
1048
std::vector<sycl::event> ind_pack_depends{packed_ind_ptrs_copy_ev,
0 commit comments