From 797ba3f7cdee6acf961747f8f42bc35748d276c2 Mon Sep 17 00:00:00 2001 From: Mohammed Suhail Munshi Date: Fri, 3 Feb 2023 11:30:52 +0000 Subject: Fix armv7a failing GEMMConvolutionLayer tests Resolves: [COMPMID-5854] Signed-off-by: Mohammed Suhail Munshi Change-Id: Ib0228409be5e816acca7e123f2660eb01a79e38f Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9078 Benchmark: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: Gunes Bayir Comments-Addressed: Arm Jenkins --- src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp b/src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp index c69af55f57..190487eced 100644 --- a/src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp +++ b/src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp @@ -839,7 +839,7 @@ Status validate_arguments(const ITensorInfo *mm_result, const ITensorInfo *vecto } // Check Tensor Rank of vector_sum_row - ARM_COMPUTE_RETURN_ERROR_ON(vector_sum_row->num_dimensions() > 2); + ARM_COMPUTE_RETURN_ERROR_ON(vector_sum_row->num_dimensions() > 3); } if(output->total_size() != 0) -- cgit v1.2.1