Skip to content

Commit 4138d8a

Browse files
committed
Use std::equal and get_size instead of loop over shape
1 parent b0dbc94 commit 4138d8a

File tree

1 file changed

+3
-8
lines changed

1 file changed

+3
-8
lines changed

dpnp/backend/extensions/indexing/choose.cpp

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -187,14 +187,9 @@ std::pair<sycl::event, sycl::event>
187187
const py::ssize_t *dst_shape = dst.get_shape_raw();
188188
const py::ssize_t *chc_shape = chc_rep.get_shape_raw();
189189

190-
bool shapes_equal(true);
191-
size_t nelems(1);
192-
for (int i = 0; i < nd; ++i) {
193-
auto src_shape_i = src_shape[i];
194-
nelems *= static_cast<size_t>(src_shape_i);
195-
shapes_equal = shapes_equal && (src_shape_i == dst_shape[i]) &&
196-
(src_shape_i == chc_shape[i]);
197-
}
190+
size_t nelems = src.get_size();
191+
bool shapes_equal = std::equal(src_shape, src_shape + nd, dst_shape);
192+
shapes_equal &= std::equal(src_shape, src_shape + nd, chc_shape);
198193

199194
if (!shapes_equal) {
200195
throw py::value_error("Array shapes don't match.");

0 commit comments

Comments
 (0)