We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 70ec4d7 commit f0e6afcCopy full SHA for f0e6afc
1 file changed
dpctl/tensor/libtensor/source/triul_ctor.cpp
@@ -229,11 +229,11 @@ usm_ndarray_triul(sycl::queue exec_q,
229
(*shp_host_shape_and_strides)[3 * nd - 1] = dst_strides[src_nd - 1];
230
231
py::ssize_t *dev_shape_and_strides =
232
- sycl::malloc_device<ssize_t>(3 * nd, exec_q);
+ sycl::malloc_device<py::ssize_t>(3 * nd, exec_q);
233
if (dev_shape_and_strides == nullptr) {
234
throw std::runtime_error("Unabled to allocate device memory");
235
}
236
- sycl::event copy_shape_and_strides = exec_q.copy<ssize_t>(
+ sycl::event copy_shape_and_strides = exec_q.copy<py::ssize_t>(
237
shp_host_shape_and_strides->data(), dev_shape_and_strides, 3 * nd);
238
239
py::ssize_t inner_range = src_shape[src_nd - 1] * src_shape[src_nd - 2];
0 commit comments