From b2a3cb2c4fcefde322eb3e8eb0c37abbdb14eb8b Mon Sep 17 00:00:00 2001 From: Anton Volkov Date: Wed, 4 Jun 2025 18:48:06 -0700 Subject: [PATCH] Pass dependent events to copy() call --- dpctl/tensor/libtensor/source/copy_for_reshape.cpp | 2 +- dpctl/tensor/libtensor/source/copy_for_roll.cpp | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/dpctl/tensor/libtensor/source/copy_for_reshape.cpp b/dpctl/tensor/libtensor/source/copy_for_reshape.cpp index eb404a4543..5eb0809768 100644 --- a/dpctl/tensor/libtensor/source/copy_for_reshape.cpp +++ b/dpctl/tensor/libtensor/source/copy_for_reshape.cpp @@ -108,7 +108,7 @@ copy_usm_ndarray_for_reshape(const dpctl::tensor::usm_ndarray &src, const char *src_data = src.get_data(); char *dst_data = dst.get_data(); sycl::event copy_ev = - exec_q.copy(src_data, dst_data, src_elemsize); + exec_q.copy(src_data, dst_data, src_elemsize, depends); return std::make_pair(keep_args_alive(exec_q, {src, dst}, {copy_ev}), copy_ev); } diff --git a/dpctl/tensor/libtensor/source/copy_for_roll.cpp b/dpctl/tensor/libtensor/source/copy_for_roll.cpp index ef63060b4f..9bac9db768 100644 --- a/dpctl/tensor/libtensor/source/copy_for_roll.cpp +++ b/dpctl/tensor/libtensor/source/copy_for_roll.cpp @@ -132,7 +132,7 @@ copy_usm_ndarray_for_roll_1d(const dpctl::tensor::usm_ndarray &src, const char *src_data = src.get_data(); char *dst_data = dst.get_data(); sycl::event copy_ev = - exec_q.copy(src_data, dst_data, src_elemsize); + exec_q.copy(src_data, dst_data, src_elemsize, depends); return std::make_pair(keep_args_alive(exec_q, {src, dst}, {copy_ev}), copy_ev); } @@ -282,7 +282,7 @@ copy_usm_ndarray_for_roll_nd(const dpctl::tensor::usm_ndarray &src, // typenames must be the same if (src_typenum != dst_typenum) { throw py::value_error( - "copy_usm_ndarray_for_reshape requires src and dst to " + "copy_usm_ndarray_for_roll_nd requires src and dst to " "have the same type."); } @@ -304,7 +304,7 @@ copy_usm_ndarray_for_roll_nd(const dpctl::tensor::usm_ndarray &src, const char *src_data = src.get_data(); char *dst_data = dst.get_data(); sycl::event copy_ev = - exec_q.copy(src_data, dst_data, src_elemsize); + exec_q.copy(src_data, dst_data, src_elemsize, depends); return std::make_pair(keep_args_alive(exec_q, {src, dst}, {copy_ev}), copy_ev); }