commit | fbe94da93b5be8745727ba7624b3d011e2bfa383 | [log] [tgz] |
---|---|---|
author | Mohammed Suhail Munshi <MohammedSuhail.Munshi@arm.com> | Fri Feb 03 11:30:52 2023 +0000 |
committer | Mohmun02 <MohammedSuhail.Munshi@arm.com> | Fri Feb 03 14:22:49 2023 +0000 |
tree | 4e5cd633b601853538e0ed4f3c13f8717592dbe6 | |
parent | d04528bcf61976225fd135efd9f5f76dc12f374d [diff] [blame] |
Fix armv7a failing GEMMConvolutionLayer tests Resolves: [COMPMID-5854] Signed-off-by: Mohammed Suhail Munshi <MohammedSuhail.Munshi@arm.com> Change-Id: Ib0228409be5e816acca7e123f2660eb01a79e38f Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9078 Benchmark: Arm Jenkins <bsgcomp@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Gunes Bayir <gunes.bayir@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
diff --git a/src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp b/src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp index c69af55..190487e 100644 --- a/src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp +++ b/src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp
@@ -839,7 +839,7 @@ } // 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)