Skip to content

Commit 7601dc5

Browse files
committed
Resolves #1327 by fixing a typo in the boolean reduction Python binding
1 parent a1dd350 commit 7601dc5

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

dpctl/tensor/libtensor/source/boolean_reductions.hpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -235,7 +235,8 @@ py_boolean_reduction(dpctl::tensor::usm_ndarray src,
235235
std::get<2>(iter_red_metadata_packing_triple_);
236236

237237
py::ssize_t *iter_shape_and_strides = packed_shapes_and_strides;
238-
py::ssize_t *red_shape_stride = packed_shapes_and_strides + (3 * iter_nd);
238+
py::ssize_t *red_shape_stride =
239+
packed_shapes_and_strides + 3 * simplified_iter_shape.size();
239240

240241
std::vector<sycl::event> all_deps;
241242
all_deps.reserve(depends.size() + 1);
@@ -244,7 +245,7 @@ py_boolean_reduction(dpctl::tensor::usm_ndarray src,
244245
all_deps.push_back(copy_metadata_ev);
245246

246247
auto red_ev =
247-
fn(exec_q, dst_nelems, red_nelems, src_data, dst_data, dst_nd,
248+
fn(exec_q, dst_nelems, red_nelems, src_data, dst_data, iter_nd,
248249
iter_shape_and_strides, iter_src_offset, iter_dst_offset,
249250
simplified_red_nd, red_shape_stride, red_src_offset, all_deps);
250251

0 commit comments

Comments
 (0)