Skip to content

Commit a26cac1

Browse files
authored
Merge pull request #2095 from IntelPython/add-missing-dep-events-to-copy-call
Add passing dependent events to `copy` calls
2 parents c85cc66 + b2a3cb2 commit a26cac1

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

dpctl/tensor/libtensor/source/copy_for_reshape.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,7 @@ copy_usm_ndarray_for_reshape(const dpctl::tensor::usm_ndarray &src,
108108
const char *src_data = src.get_data();
109109
char *dst_data = dst.get_data();
110110
sycl::event copy_ev =
111-
exec_q.copy<char>(src_data, dst_data, src_elemsize);
111+
exec_q.copy<char>(src_data, dst_data, src_elemsize, depends);
112112
return std::make_pair(keep_args_alive(exec_q, {src, dst}, {copy_ev}),
113113
copy_ev);
114114
}

dpctl/tensor/libtensor/source/copy_for_roll.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,7 @@ copy_usm_ndarray_for_roll_1d(const dpctl::tensor::usm_ndarray &src,
132132
const char *src_data = src.get_data();
133133
char *dst_data = dst.get_data();
134134
sycl::event copy_ev =
135-
exec_q.copy<char>(src_data, dst_data, src_elemsize);
135+
exec_q.copy<char>(src_data, dst_data, src_elemsize, depends);
136136
return std::make_pair(keep_args_alive(exec_q, {src, dst}, {copy_ev}),
137137
copy_ev);
138138
}
@@ -282,7 +282,7 @@ copy_usm_ndarray_for_roll_nd(const dpctl::tensor::usm_ndarray &src,
282282
// typenames must be the same
283283
if (src_typenum != dst_typenum) {
284284
throw py::value_error(
285-
"copy_usm_ndarray_for_reshape requires src and dst to "
285+
"copy_usm_ndarray_for_roll_nd requires src and dst to "
286286
"have the same type.");
287287
}
288288

@@ -304,7 +304,7 @@ copy_usm_ndarray_for_roll_nd(const dpctl::tensor::usm_ndarray &src,
304304
const char *src_data = src.get_data();
305305
char *dst_data = dst.get_data();
306306
sycl::event copy_ev =
307-
exec_q.copy<char>(src_data, dst_data, src_elemsize);
307+
exec_q.copy<char>(src_data, dst_data, src_elemsize, depends);
308308
return std::make_pair(keep_args_alive(exec_q, {src, dst}, {copy_ev}),
309309
copy_ev);
310310
}

0 commit comments

Comments
 (0)