diff options
author | Mohammed Suhail Munshi <MohammedSuhail.Munshi@arm.com> | 2023-02-03 11:30:52 +0000 |
---|---|---|
committer | Mohmun02 <MohammedSuhail.Munshi@arm.com> | 2023-02-03 14:22:49 +0000 |
commit | fbe94da93b5be8745727ba7624b3d011e2bfa383 (patch) | |
tree | 4e5cd633b601853538e0ed4f3c13f8717592dbe6 | |
parent | d04528bcf61976225fd135efd9f5f76dc12f374d (diff) | |
download | ComputeLibrary-fbe94da93b5be8745727ba7624b3d011e2bfa383.tar.gz |
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>
-rw-r--r-- | src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
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) |