File tree Expand file tree Collapse file tree 1 file changed +3
-8
lines changed
dpnp/backend/extensions/indexing Expand file tree Collapse file tree 1 file changed +3
-8
lines changed Original file line number Diff line number Diff 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." );
You can’t perform that action at this time.
0 commit comments