diff options
Diffstat (limited to 'src/gpu/cl/kernels/helpers')
| -rw-r--r-- | src/gpu/cl/kernels/helpers/MatMulKernelHelpers.cpp | 15 | ||||
| -rw-r--r-- | src/gpu/cl/kernels/helpers/MatMulKernelHelpers.h | 12 |
2 files changed, 18 insertions, 9 deletions
diff --git a/src/gpu/cl/kernels/helpers/MatMulKernelHelpers.cpp b/src/gpu/cl/kernels/helpers/MatMulKernelHelpers.cpp index 2407c6ca5e..689a743fdf 100644 --- a/src/gpu/cl/kernels/helpers/MatMulKernelHelpers.cpp +++ b/src/gpu/cl/kernels/helpers/MatMulKernelHelpers.cpp @@ -36,7 +36,9 @@ namespace opencl { namespace kernels { -Status validate_matmul_input_shapes(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const MatMulKernelInfo &matmul_kernel_info) +Status validate_matmul_input_shapes(const TensorShape &lhs_shape, + const TensorShape &rhs_shape, + const MatMulKernelInfo &matmul_kernel_info) { const size_t lhs_k = matmul_kernel_info.adj_lhs ? lhs_shape.y() : lhs_shape.x(); const size_t rhs_k = matmul_kernel_info.adj_rhs ? rhs_shape.x() : rhs_shape.y(); @@ -46,7 +48,7 @@ Status validate_matmul_input_shapes(const TensorShape &lhs_shape, const TensorSh ARM_COMPUTE_RETURN_ERROR_ON_MSG(rhs_shape.total_size() == 0, "Rhs tensor can't be empty"); constexpr size_t batch_dim_start = 2; - for(size_t i = batch_dim_start; i < Coordinates::num_max_dimensions; ++i) + for (size_t i = batch_dim_start; i < Coordinates::num_max_dimensions; ++i) { ARM_COMPUTE_RETURN_ERROR_ON_MSG(lhs_shape[i] != rhs_shape[i], "Batch dimension broadcasting is not supported"); } @@ -54,9 +56,12 @@ Status validate_matmul_input_shapes(const TensorShape &lhs_shape, const TensorSh return Status{}; } -std::pair<Status, Window> validate_and_configure_window_for_mmul_kernels(const ITensorInfo *lhs, - const ITensorInfo *rhs, const ITensorInfo *dst, const MatMulKernelInfo &matmul_kernel_info, - int mmul_m0, int mmul_n0) +std::pair<Status, Window> validate_and_configure_window_for_mmul_kernels(const ITensorInfo *lhs, + const ITensorInfo *rhs, + const ITensorInfo *dst, + const MatMulKernelInfo &matmul_kernel_info, + int mmul_m0, + int mmul_n0) { ARM_COMPUTE_UNUSED(lhs, rhs); diff --git a/src/gpu/cl/kernels/helpers/MatMulKernelHelpers.h b/src/gpu/cl/kernels/helpers/MatMulKernelHelpers.h index 210f22b109..c2ae2a67f4 100644 --- a/src/gpu/cl/kernels/helpers/MatMulKernelHelpers.h +++ b/src/gpu/cl/kernels/helpers/MatMulKernelHelpers.h @@ -44,7 +44,8 @@ namespace kernels * * @return true if the shapes and matmul kernel info matches */ -Status validate_matmul_input_shapes(const TensorShape &lhs_shape, const TensorShape &rhs_shape, +Status validate_matmul_input_shapes(const TensorShape &lhs_shape, + const TensorShape &rhs_shape, const MatMulKernelInfo &matmul_kernel_info); /** Validate and configure window for Matmul MMUL kernels @@ -58,9 +59,12 @@ Status validate_matmul_input_shapes(const TensorShape &lhs_shape, const TensorSh * * @return a pair of Status and Window object */ -std::pair<Status, Window> validate_and_configure_window_for_mmul_kernels(const ITensorInfo *lhs, - const ITensorInfo *rhs, const ITensorInfo *dst, const MatMulKernelInfo &matmul_kernel_info, - int mmul_m0, int mmul_n0); +std::pair<Status, Window> validate_and_configure_window_for_mmul_kernels(const ITensorInfo *lhs, + const ITensorInfo *rhs, + const ITensorInfo *dst, + const MatMulKernelInfo &matmul_kernel_info, + int mmul_m0, + int mmul_n0); } // namespace kernels } // namespace opencl } // namespace arm_compute |