@@ -609,34 +609,31 @@ std::pair<sycl::event, sycl::event>
609609 shT outer_inner_x1_shape;
610610 shT batch_x1_strides;
611611 shT outer_inner_x1_strides;
612- split_iteration_space (
613- x1_shape_vec, x1_strides_vec, batch_dims,
614- batch_dims + x1_outer_inner_dims,
615- // 4 vectors modified
616- batch_x1_shape, outer_inner_x1_shape, batch_x1_strides,
617- outer_inner_x1_strides);
612+ split_iteration_space (x1_shape_vec, x1_strides_vec, batch_dims,
613+ batch_dims + x1_outer_inner_dims,
614+ // 4 vectors modified
615+ batch_x1_shape, outer_inner_x1_shape,
616+ batch_x1_strides, outer_inner_x1_strides);
618617
619618 shT batch_x2_shape;
620619 shT outer_inner_x2_shape;
621620 shT batch_x2_strides;
622621 shT outer_inner_x2_strides;
623- split_iteration_space (
624- x2_shape_vec, x2_strides_vec, batch_dims,
625- batch_dims + x2_outer_inner_dims,
626- // 4 vectors modified
627- batch_x2_shape, outer_inner_x2_shape, batch_x2_strides,
628- outer_inner_x2_strides);
622+ split_iteration_space (x2_shape_vec, x2_strides_vec, batch_dims,
623+ batch_dims + x2_outer_inner_dims,
624+ // 4 vectors modified
625+ batch_x2_shape, outer_inner_x2_shape,
626+ batch_x2_strides, outer_inner_x2_strides);
629627
630628 shT batch_dst_shape;
631629 shT outer_inner_dst_shape;
632630 shT batch_dst_strides;
633631 shT outer_inner_dst_strides;
634- split_iteration_space (
635- dst_shape_vec, dst_strides_vec, batch_dims,
636- batch_dims + dst_outer_inner_dims,
637- // 4 vectors modified
638- batch_dst_shape, outer_inner_dst_shape, batch_dst_strides,
639- outer_inner_dst_strides);
632+ split_iteration_space (dst_shape_vec, dst_strides_vec, batch_dims,
633+ batch_dims + dst_outer_inner_dims,
634+ // 4 vectors modified
635+ batch_dst_shape, outer_inner_dst_shape,
636+ batch_dst_strides, outer_inner_dst_strides);
640637
641638 using shT = std::vector<py::ssize_t >;
642639 shT simplified_batch_shape;
@@ -834,4 +831,4 @@ void init_dot(py::module_ m)
834831 m.def (" _dot_result_type" , dot_result_type_pyapi, " " );
835832}
836833
837- } // namespace dpctl::tensor::py_internal
834+ } // namespace dpctl::tensor::py_internal
0 commit comments