Skip to content

Commit d5a49c2

Browse files
Reordered waits
1 parent 7fee9e4 commit d5a49c2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dpctl/tensor/libtensor/source/integer_advanced_indexing.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -723,8 +723,8 @@ usm_ndarray_take(dpctl::tensor::usm_ndarray src,
723723
});
724724
});
725725

726-
sycl::event::wait(host_task_events);
727726
sycl::event::wait({take_generic_ev, temporaries_cleanup_ev});
727+
sycl::event::wait(host_task_events);
728728

729729
/*
730730
sycl::event host_task_ev = keep_args_alive(exec_q, {src, py_ind, dst},
@@ -1137,8 +1137,8 @@ usm_ndarray_put(dpctl::tensor::usm_ndarray dst,
11371137
});
11381138
});
11391139

1140-
sycl::event::wait(host_task_events);
11411140
sycl::event::wait({put_generic_ev, temporaries_cleanup_ev});
1141+
sycl::event::wait(host_task_events);
11421142

11431143
/*
11441144
sycl::event py_obj_cleanup_ev =

0 commit comments

Comments
 (0)